--n_epochs = 1000
--patience = 10
--mbatch_size = 16
--lr = 0.001
--lr_scheduler = "StepLR"
--lr_scheduler_kwargs = {"step_size": 5, "gamma": 0.7}
--param_lr_map = {}
--grad_clip = null
--grad_clip_kwargs = {}
--regularization = 0.0
--l1_reg = 0.0
--l2_reg = 0.0
--optimizer = "Adam"
--loss = "MSELoss"
--initializer = null
--initialization_seed = 0
--batch_shuffle_seed = 0