From 794d12bd5c43e230b36554d8d484757ed7942583 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Wed, 19 Jun 2024 15:59:51 -0400 Subject: [PATCH] Fix detectron2 maskrcnn --- torchbenchmark/models/detectron2_maskrcnn/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchbenchmark/models/detectron2_maskrcnn/__init__.py b/torchbenchmark/models/detectron2_maskrcnn/__init__.py index 2d5753e32..3a42d5797 100644 --- a/torchbenchmark/models/detectron2_maskrcnn/__init__.py +++ b/torchbenchmark/models/detectron2_maskrcnn/__init__.py @@ -58,6 +58,7 @@ class Model(BenchmarkModel): model_file = os.path.join(MODEL_DIR, ".data", f"{MODEL_NAME}.pkl") DEFAULT_TRAIN_BSIZE = 1 DEFAULT_EVAL_BSIZE = 1 + DISABLE_DETERMINISM = True def __init__(self, test, device, batch_size=None, extra_args=[]): super().__init__(