search

facebookarchive / fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts
Other
2.29k stars 664 forks source link

Fix OOM errors when reloading certain large checkpointed models #161

Closed lvdmaaten closed 7 years ago

lvdmaaten commented 7 years ago

PR https://github.com/facebook/fb.resnet.torch/pull/129 fixed an OOM issue when reloading large models via retrain, but forgot to do it when reloading checkpoints. This PR corrects that.