Rename variables

dinhanhx · Sep 23, 2022 · 1564c23 · 1564c23
1 parent 07878fe
commit 1564c23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run_uit_viic.py b/run_uit_viic.py
@@ -75,12 +75,12 @@ def configure_optimizers(self):
                                           image_size=config.image_size,
                                           patch_size=config.patch_size)
 
-    sampler_train = DistributedSampler(
+    sampler = DistributedSampler(
             train_val_ic, num_replicas=xm.xrt_world_size(), rank=xm.get_ordinal(), shuffle=False
         )
     dataloader = DataLoader(train_val_ic,
                             batch_size=8,
-                            sampler=sampler_train,
+                            sampler=sampler,
                             num_workers=24,
                             collate_fn=ic_collator,
                             drop_last=True)