mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-20 13:50:41 +00:00
16 lines
231 B
Python
16 lines
231 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]
|
|
PRECISION = 16
|