diff --git a/projects/implicitron_trainer/experiment.py b/projects/implicitron_trainer/experiment.py index c042e965..a033251a 100755 --- a/projects/implicitron_trainer/experiment.py +++ b/projects/implicitron_trainer/experiment.py @@ -83,6 +83,9 @@ from .impl.utils import seed_all_random_engines logger = logging.getLogger(__name__) +# workaround for https://github.com/facebookresearch/hydra/issues/2262 +_RUN = hydra.types.RunMode.RUN + if version.parse(hydra.__version__) < version.Version("1.1"): raise ValueError( f"Hydra version {hydra.__version__} is too old." @@ -136,6 +139,7 @@ class Experiment(Configurable): # pyre-ignore: 13 default_factory=lambda: { "run": {"dir": "."}, # Make hydra not change the working dir. "output_subdir": None, # disable storing the .hydra logs + "mode": _RUN, } ) diff --git a/projects/implicitron_trainer/tests/experiment.yaml b/projects/implicitron_trainer/tests/experiment.yaml index 976027ca..9588043b 100644 --- a/projects/implicitron_trainer/tests/experiment.yaml +++ b/projects/implicitron_trainer/tests/experiment.yaml @@ -9,6 +9,7 @@ hydra: run: dir: . output_subdir: null + mode: RUN data_source_ImplicitronDataSource_args: dataset_map_provider_class_type: ??? data_loader_map_provider_class_type: SequenceDataLoaderMapProvider