mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-21 11:18:01 +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
|