Corrected due to the following error

ValueError: Default process group has not been initialized, please make sure to call init_process_group facebookresearch#55
saten-private · Jun 13, 2024 · f9af285 · f9af285
1 parent c1cda69
commit f9af285
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/evals/video_classification_frozen/eval.py b/evals/video_classification_frozen/eval.py
@@ -228,7 +228,10 @@ def main(args_eval, resume_preempt=False):
         warmup=warmup,
         num_epochs=num_epochs,
         use_bfloat16=use_bfloat16)
-    classifier = DistributedDataParallel(classifier, static_graph=True)
+    # Corrected due to the following error
+    # ValueError: Default process group has not been initialized, please make sure to call init_process_group
+    # https://github.com/facebookresearch/jepa/issues/55
+    # classifier = DistributedDataParallel(classifier, static_graph=True)
 
     # -- load training checkpoint
     start_epoch = 0