diff --git a/README.md b/README.md index b8b7d059a20529f873c64626be8d04c917ad34bb..246b1867768743edfffddc5fa9ea36179d01f961 100644 --- a/README.md +++ b/README.md @@ -19,11 +19,11 @@ poetry install Once the project has been installed, you can run it with the following command: ``` -poetry run flashtrain +python -m hydra_flash.train ``` You can, for example, change the learning rate using : ``` -poetry run flashtrain model.learning_rate=1e-2 +python -m hydra_flash.train model.learning_rate=1e-2 ``` diff --git a/hydra_flash/train.py b/hydra_flash/train.py index 8af9f8157d372f0a40179d1c771df5b8b383aad7..fa5aa5faa13d7b59efbd505bc009b016e8fb0291 100644 --- a/hydra_flash/train.py +++ b/hydra_flash/train.py @@ -23,3 +23,7 @@ def main(cfg): predict_datamodule = instantiate(cfg.predict_datamodule) predictions = trainer.predict(model, datamodule=predict_datamodule, output="labels") log.info(f"predictions : {predictions}") + + +if __name__ == "__main__": + main()