| commit | b71cebb11ffc536c59e68586d635f39d5b26fa77 | [log] [tgz] |
|---|---|---|
| author | Aapo Kyrola <akyrola@fb.com> | Fri Nov 03 10:58:01 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Fri Nov 03 11:15:25 2017 -0700 |
| tree | cc0cafc20bc8c7f3130205595812666b99b523cb | |
| parent | 1b5c843a9c2b8e96398245b3a794df74d923f94d [diff] |
Fix LoadModel() in resnet50_trainer Summary: resnet50 trainer will save the 'optimizer_iteration' blob in checkpoints, but loads it i in GPU context. This fails because AtomicIter/Iter expect the blob to be in CPU context. So manually reset the optimizer_iteration in CPU context. I am thinking of making the iter-operators automatically do this switch, but in the mean time this unbreaks the trainer. Reviewed By: sf-wind Differential Revision: D6232626 fbshipit-source-id: da7c183a87803e008f94c86b6574b879c3b76438
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the Apache 2.0 license. See the NOTICE file for details.