求助出现MemoryError怎么解决

haiou_f · 发表于 2022-3-18 19:31:52

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

平时训练bs可以到12，然后把bs调到6，颜色模式从none改为rct，每次保存就出现如下提示，请问该怎么设置？

保存时间|迭代次数|单次时间|SRC损失|DST损失
Error:
Traceback (most recent call last):
  File "E:\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\mainscripts\Trainer.py", line 198, in trainerThread
model_save()
  File "E:\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\mainscripts\Trainer.py", line 68, in model_save
model.save()
  File "E:\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\models\ModelBase.py", line 399, in save
self.onSave()
  File "E:\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\models\Model_AMP\Model.py", line 638, in onSave
model.save_weights ( self.get_strpath_storage_for_file(filename) )
  File "E:\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\core\leras\layers\Saveable.py", line 61, in save_weights
d_dumped = pickle.dumps (d, 4)
MemoryError