From 5bf9235967a8cd7b532344ffd7bbe38d1bf2a2ad Mon Sep 17 00:00:00 2001 From: Iason Myttas Date: Tue, 9 Jan 2024 16:01:28 +0000 Subject: [PATCH] added requirements for 3 models --- torchbenchmark/models/drq/requirements.txt | 1 + torchbenchmark/models/soft_actor_critic/requirements.txt | 1 + torchbenchmark/models/vision_maskrcnn/requirements.txt | 1 + 3 files changed, 3 insertions(+) diff --git a/torchbenchmark/models/drq/requirements.txt b/torchbenchmark/models/drq/requirements.txt index 9afa868267..cfbd5fcdd6 100644 --- 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 index e844e24ccf..9dd7425d1e 100644 --- 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 index e69de29bb2..99f22cd125 100644 --- a/torchbenchmark/models/vision_maskrcnn/requirements.txt +++ b/torchbenchmark/models/vision_maskrcnn/requirements.txt @@ -0,0 +1 @@ +pycocotools