Closed chenrxi closed 2 years ago
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.
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.