Fix detectron2 maskrcnn

xuzhao9 · xuzhao9 · commit 794d12bd5c43 · 2024-06-19T15:59:51.000-04:00
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__(
