OPTIMIZER = 'optimizer'
MODEL = 'model'
CRITERION = 'criterion'
LOADER = 'loader'
MODEL_CLASS = 'model_class'
SHOULD_TRAIN_MODEL = 'should_train'
MODEL_PATH = 'model_path'
TRAIN_STEP = 'train_step'
DATASET_NAME = 'dataset_name'
LEARNING_RATE = "learning_rate"
BATCH_SIZE = 'batch_size'
NUM_EPOCHS = 'num_epochs'
EPOCHS_TO_SAVE = 'epochs_to_save'
MODEL_SAVE_DIR = 'model_save_dir'
ENSEMBLE_NAME = 'ensemble_name'
SLURM_TASK_ID = 'slurm_task_id'
NUM_LOADER_WORKERS = 'num_loader_workers'
ENSEMBLE = 'ensemble'
DEVICE = 'device'
TEST_ENSEMBLE = 'test_ensemble'
SAVED_PATH = 'saved_path'