import torch

# Training Hyperparameters
LEARNING_RATE     = 0.0001
BATCH_SIZE        = 16
NUM_WORKERS       = 24
MAX_EPOCHS        = 50

# Dataset parameters
WEIGHTS           = None
SPLIT_BY          = 'sequence'
LOAD_DATA         = 'all_data'
DATASET_SPLIT     = (0.8, 0.2, 0)
STANDARDIZE_RANGE = (170, 300)
DOWNSAMPLE_SIZE   = 224 # Overwritten if size argument is given
NUM_CLASSES       = 5

# Computation parameters
ACCELERATOR       = 'gpu' if torch.cuda.is_available() else 'cpu'
DEVICES           = 0 # Overwritten if device argument is given
DATA_DIR          = '/data/'