Tensorboard Version Fix (facebookresearch#874)

Bigazzon · May 24, 2022 · b54dd6e · b54dd6e
1 parent 323f786
commit b54dd6e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/habitat_baselines/rl/requirements.txt b/habitat_baselines/rl/requirements.txt
@@ -1,3 +1,4 @@
 moviepy>=1.0.1
 torch>=1.3.1
-tensorboard
+protobuf==3.20.1
+tensorboard==2.8.0
diff --git a/test/test_baseline_trainers.py b/test/test_baseline_trainers.py
@@ -78,7 +78,7 @@ def _powerset(s):
 )
 def test_trainers(test_cfg_path, mode, gpu2gpu, observation_transforms):
     # For testing with world_size=1, -1 works as port in PyTorch
-    os.environ["MASTER_PORT"] = str(-1)
+    os.environ["MAIN_PORT"] = str(-1)
 
     if gpu2gpu:
         try: