| 55 | } |
| 56 | |
| 57 | virtual void CopyNetParams(const bool copy_diff, |
| 58 | vector<shared_ptr<Blob<Dtype> > >* params_copy) { |
| 59 | CHECK(net_); |
| 60 | const vector<shared_ptr<Blob<Dtype> > >& net_params = net_->params(); |
| 61 | params_copy->clear(); |
| 62 | params_copy->resize(net_params.size()); |
| 63 | const bool kReshape = true; |
| 64 | for (int i = 0; i < net_params.size(); ++i) { |
| 65 | (*params_copy)[i].reset(new Blob<Dtype>()); |
| 66 | (*params_copy)[i]->CopyFrom(*net_params[i], copy_diff, kReshape); |
| 67 | } |
| 68 | } |
| 69 | |
| 70 | virtual void InitTinyNet(const bool force_backward = false, |
| 71 | const bool accuracy_layer = false) { |
no test coverage detected