diff options
Diffstat (limited to 'experiment/__main__.py')
-rw-r--r-- | experiment/__main__.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/experiment/__main__.py b/experiment/__main__.py new file mode 100644 index 0000000..bf4de94 --- /dev/null +++ b/experiment/__main__.py @@ -0,0 +1,17 @@ +import sys + +import torch + +from experiment.experiment import Experiment + +file = None +if len(sys.argv) < 2: + print('Provide YAML configuration file as argument') + exit(1) + +device = torch.device("cuda" if torch.cuda.is_available() else "cpu") +file = sys.argv[1] +prefix = sys.argv[2] if len(sys.argv) > 2 else '' + +experiment = Experiment(file, prefix, device) +experiment.run() |