(self)
| 36 | return data_set, data_loader |
| 37 | |
| 38 | def _select_optimizer(self): |
| 39 | # model_optim = optim.Adam(self.model.parameters(), lr=self.args.learning_rate) |
| 40 | model_optim = optim.RAdam(self.model.parameters(), lr=self.args.learning_rate) |
| 41 | return model_optim |
| 42 | |
| 43 | def _select_criterion(self): |
| 44 | criterion = nn.CrossEntropyLoss() |