From b9da26e92ddaaeb9429806bcab3c30bb4db6b19f Mon Sep 17 00:00:00 2001 From: Iason Myttas Date: Fri, 19 Jan 2024 15:06:04 -0800 Subject: [PATCH] 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: https://github.com/pytorch/benchmark/pull/2115 Reviewed By: aaronenyeshi Differential Revision: D52923310 Pulled By: xuzhao9 fbshipit-source-id: d984dbdbb025f3ab25a3611601d2a3564751545f --- 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