From 6633c2a8c0a3818ba68f0089b9c91163bae3a7dd Mon Sep 17 00:00:00 2001
From: Victor Joos <victor.joos@uclouvain.be>
Date: Mon, 27 Jun 2022 14:07:02 +0200
Subject: [PATCH] Adds regular python launcher instead of poetry run
---
README.md | 4 ++--
hydra_flash/train.py | 4 ++++
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/README.md b/README.md
index b8b7d05..246b186 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 8af9f81..fa5aa5f 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()
--
GitLab