Fixes missing DataParallel in rs train

Author: daniel-j-h

robosat/tools/train.py

@@ -63,7 +63,9 @@ def main(args):
     os.makedirs(model["common"]["checkpoint"], exist_ok=True)
 
     num_classes = len(dataset["common"]["classes"])
-    net = FPNSegmentation(num_classes).to(device)
+    net = FPNSegmentation(num_classes)
+    net = DataParallel(net)
+    net = net.to(device)
 
     if model["common"]["cuda"]:
         torch.backends.cudnn.benchmark = True