diff --git a/torchbenchmark/models/functorch_maml_omniglot/install.py b/torchbenchmark/models/functorch_maml_omniglot/install.py index be308ead48..d3e20cdb9b 100644 --- a/torchbenchmark/models/functorch_maml_omniglot/install.py +++ b/torchbenchmark/models/functorch_maml_omniglot/install.py @@ -1,5 +1,6 @@ import subprocess import sys +from utils import s3_utils def pip_install_requirements(): @@ -7,3 +8,4 @@ def pip_install_requirements(): if __name__ == '__main__': pip_install_requirements() + s3_utils.checkout_s3_data("MODEL_PKLS", "maml_omniglot/batch.pt", decompress=False)