File size: 581 Bytes
1964059 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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/' |