--n_epochs = 50
--patience = -1
--mbatch_size = 128
--lr = 0.001
--lr_scheduler = null
--lr_scheduler_kwargs = {}
--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
--initialization = null