lif31up's picture
change
a7f90b0
raw
history blame
163 Bytes
HYPERPARAMETER_CONFIG = {
"lr": 0.0001,
"weight_decay": 0.0001
} # HYPERPARAMETER_CONFIG
TRAINING_CONFIG = {
"iters": 10,
"epochs": 30,
} # TRAINING_CONFIG