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__(