(storage, location)
| 139 | def load_state(path, model, ignore=[], optimizer=None, cuda=False, recover=False, |
| 140 | remove_prefix=None, strict=False): |
| 141 | def map_func_cuda(storage, location): |
| 142 | return storage.cuda() |
| 143 | def map_func_cpu(storage, location): |
| 144 | return storage.cpu() |
| 145 | if cuda: |