Jingyu Kong

Results 2 issues of Jingyu Kong

When I trained with the previous version, the code worked fine when the dataset size was 256, but I got an error when my dataset size was 512.How should I...

When I use --multi_gpu 0,1,2, it has a error: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! And how...