DefaultCPUAllocator: not enough memory: you tried to allocate 94633984 bytes. Buy new RAM!

问题描述

在复现Tianxiaomo/pytorch-YOLOv4的时候出现如下错误
在这里插入图片描述
问题就是cpu内存不够

解决方案

  1. 换用gpu做训练
  2. 减小epoch的值,
  3. 减小batch_size(2的倍数)的值

你可能感兴趣的:(python错误解决,python)