HYPERPARAMETER_CONFIG = {
  "lr": 0.0001,
  "weight_decay": 0.0001
} # HYPERPARAMETER_CONFIG
TRAINING_CONFIG = {
  "iters": 10,
  "epochs": 30,
} # TRAINING_CONFIG