added requirements for 3 models

pytorch · Jan 9, 2024 · 5bf9235 · 5bf9235
1 parent fce6c89
commit 5bf9235
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