|
楼主 |
发表于 2020-11-8 21:10:19
|
显示全部楼层
预训练改成no了,报错
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[6,512,144,144] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[node Conv2D_35 (defined at F:\NVIDIA802\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:99) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
[[node concat (defined at F:\NVIDIA802\_internal\DeepFaceLab\models\Model_SAEHD\Model.py:484) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. |
|