(self, batch)
| 355 | return loss_summary |
| 356 | |
| 357 | def parse_batch_train(self, batch): |
| 358 | input = batch["img"] |
| 359 | label = batch["label"] |
| 360 | input = input.to(self.device) |
| 361 | label = label.to(self.device) |
| 362 | return input, label |
| 363 | |
| 364 | |
| 365 | def load_model(self, directory, epoch=None): |