(state, path, step)
| 106 | return acc |
| 107 | |
| 108 | def save_state(state, path, step): |
| 109 | path, filename = os.path.split(path) |
| 110 | assert path != '' |
| 111 | if not os.path.exists(path): |
| 112 | os.makedirs(path) |
| 113 | print('saving to {}/{}_iter_{}.pth.tar'.format(path, filename, step)) |
| 114 | torch.save(state, '{}/{}_iter_{}.pth.tar'.format(path, filename, step)) |
| 115 | |
| 116 | def load_last_iter(path): |
| 117 | if os.path.isfile(path): |