(self, batch)
| 365 | return loss_summary |
| 366 | |
| 367 | def parse_batch_train(self, batch): |
| 368 | input = batch["img"] |
| 369 | label = batch["label"] |
| 370 | input = input.to(self.device) |
| 371 | label = label.to(self.device) |
| 372 | return input, label |
| 373 | |
| 374 | def load_model(self, directory, epoch=None): |
| 375 | if not directory: |