Bohao-Lee / CME

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Questions about nGPU

chenrxi opened this issue · comments

commented

if I set the ngpus to 4 and batch size to 64, the region loss will have an error.
output.size(0) is 1920. Do you have any idea to solve this error.