KumapowerLIU / CSA-inpainting

Coherent Semantic Attention for image inpainting(ICCV 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

训练问题

Dominic-ZZ opened this issue · comments

1.为什么batchsize只能设置为1?我尝试过设置为6,在3个epoch之后loss都变成了Nan,请问调大batchsize需要改代码的哪些地方?
2,GPU利用率不稳定的问题,当batchsize为1时,GPU的利用率在20%-90%之间波动,我试着调大了batchsize该问题仍然存在,请问该如何解决?修改dataloader的num_workers以及pin_memory也都无效。