Added installation requirements for three models (#2115)

Summary: The following three models were missing python package installations that cause them to fail: - vision_maskrcnn (pycocotools package) - soft_actor_critic (tensorboardX package) - drq (gym package) Pull Request resolved: #2115 Reviewed By: aaronenyeshi Differential Revision: D52923310 Pulled By: xuzhao9 fbshipit-source-id: d984dbdbb025f3ab25a3611601d2a3564751545f
pytorch · Jan 19, 2024 · b9da26e · b9da26e
1 parent 5e1008c
commit b9da26e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/torchbenchmark/models/drq/requirements.txt b/torchbenchmark/models/drq/requirements.txt
@@ -1,2 +1,3 @@
 kornia
 scikit-image
+gym
diff --git a/torchbenchmark/models/soft_actor_critic/requirements.txt b/torchbenchmark/models/soft_actor_critic/requirements.txt
@@ -1,2 +1,3 @@
 gym
 pygame
+tensorboardX
diff --git a/torchbenchmark/models/vision_maskrcnn/requirements.txt b/torchbenchmark/models/vision_maskrcnn/requirements.txt
@@ -0,0 +1 @@
+pycocotools