Fix detectron2 maskrcnn

pytorch · Jun 19, 2024 · 794d12b · 794d12b
1 parent c30d44d
commit 794d12b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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__(