[fix]

main.py

@@ -213,7 +213,7 @@ def main():
 
     cudnn.benchmark = True
     net.cuda()
-    device_ids = [0, 1]
+    device_ids = [0]
     # net = nn.DataParallel(net).to(device_ids)
     net = nn.DataParallel(net, device_ids=device_ids)
     # optimizer = nn.DataParallel(optimizer, device_ids=device_ids)