# Define params for data loader
BS = 32
WIDTH = 28
HEIGHT = 28
N_CHANNELS = 1
NUM_CLASSES = 10

# Define model params
HIDDEN_SIZE = 200
DROPOUT = 0.50

# Define training params
LR = 0.005
EPOCHS = 10

# Define logging directory
LOG_DIR = '~/Logs/'