| 836 | |
| 837 | template <typename Dtype> |
| 838 | void Net<Dtype>::ToProto(NetParameter* param, bool write_diff) const { |
| 839 | param->Clear(); |
| 840 | param->set_name(name_); |
| 841 | // Add bottom and top |
| 842 | DLOG(INFO) << "Serializing " << layers_.size() << " layers"; |
| 843 | for (int i = 0; i < layers_.size(); ++i) { |
| 844 | LayerParameter* layer_param = param->add_layer(); |
| 845 | layers_[i]->ToProto(layer_param, write_diff); |
| 846 | } |
| 847 | } |
| 848 | |
| 849 | template <typename Dtype> |
| 850 | void Net<Dtype>::ToHDF5(const string& filename, bool write_diff) const { |