indent. error

pytorch_job_wganSingleGen_ncc.yaml

@@ -40,7 +40,7 @@ spec:
                 limits:
                   nvidia.com/gpu: 1
     Worker:
-      replicas: 4
+      replicas: 2
       restartPolicy: OnFailure
       template:
         metadata:

wganSingleGen.py

@@ -274,9 +274,8 @@ def run(args):
     print('Critic trainable params:', sum(p.numel() for p in Crit_E_H.parameters() if p.requires_grad))
     print('Generator trainable params:', sum(p.numel() for p in Gen_E_H.parameters() if p.requires_grad))
     
-     if args.world_size > 1: 
-        Distributor = nn.parallel.DistributedDataParallel if use_cuda \
-            else nn.parallel.DistributedDataParallelCPU
+    if args.world_size > 1: 
+        Distributor = nn.parallel.DistributedDataParallel if use_cuda else nn.parallel.DistributedDataParallelCPU
         Crit_E_H = Distributor(Crit_E_H, device_ids=[args.local_rank], output_device=args.local_rank )
         Gen_E_H = Distributor(Gen_E_H, device_ids=[args.local_rank], output_device=args.local_rank )