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()