(opt, actions, val_loader, model, steps=None)
| 46 | return step('train', opt, actions, train_loader, model, optimizer, epoch) |
| 47 | |
| 48 | def val(opt, actions, val_loader, model, steps=None): |
| 49 | with torch.no_grad(): |
| 50 | return step('test', opt, actions, val_loader, model, steps=steps) |
| 51 | |
| 52 | def step(split, args, actions, dataLoader, model, optimizer=None, epoch=None, steps=None): |
| 53 |