mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-21 19:27:58 +00:00
16 lines
234 B
Python
16 lines
234 B
Python
# Training hyperparameters
|
|
INPUT_SIZE = 784
|
|
NUM_CLASSES = 10
|
|
LEARNING_RATE = 0.001
|
|
BATCH_SIZE = 64
|
|
NUM_EPOCHS = 3
|
|
|
|
# Dataset
|
|
DATA_DIR = "dataset/"
|
|
NUM_WORKERS = 4
|
|
|
|
# Compute related
|
|
ACCELERATOR = "gpu"
|
|
DEVICES = [0, 1]
|
|
PRECISION = 16
|