(self, flag)
| 32 | return model |
| 33 | |
| 34 | def _get_data(self, flag): |
| 35 | data_set, data_loader = data_provider(self.args, flag) |
| 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) |
no test coverage detected