(net, label, epoch, opt)
| 201 | |
| 202 | |
| 203 | def load_network(net, label, epoch, opt): |
| 204 | save_filename = '%s_net_%s.pth' % (epoch, label) |
| 205 | save_dir = os.path.join(opt.checkpoints_dir, opt.name) |
| 206 | save_path = os.path.join(save_dir, save_filename) |
| 207 | weights = torch.load(save_path) |
| 208 | net.load_state_dict(weights) |
| 209 | return net |
| 210 | |
| 211 | |
| 212 | ############################################################################### |
nothing calls this directly
no outgoing calls
no test coverage detected