- 博客(2)
- 收藏
- 关注
RSS订阅原创 RuntimeError: [enforce fail at ..\c10\core\CPUAllocator.cpp:76] data. DefaultCPUAllocato模型训练中的内存溢出问题
在 num_workers=0 的基础上，如果问题还存在，可以尝试将 Freeze_batch_size 和 Unfreeze_batch_size 都设置成一个更小的值，比如 4 或 2。一次处理的图片越多，在 CPU 端需要整理的数据量也越大。根据报错前的进度条 Batch size from DataLoader should be: 8，我们可以看到当前有效的批量大小是 8。请再次检查您的 Freeze_batch_size 和 Unfreeze_batch_size。
2025-07-21 11:34:55
593
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人