| 854 | |
| 855 | template <typename Dtype> |
| 856 | void Net<Dtype>::ToProto(NetParameter* param, bool write_diff) const { |
| 857 | param->Clear(); |
| 858 | param->set_name(name_); |
| 859 | // Add bottom and top |
| 860 | DLOG(INFO) << "Serializing " << layers_.size() << " layers"; |
| 861 | for (int i = 0; i < layers_.size(); ++i) { |
| 862 | LayerParameter* layer_param = param->add_layer(); |
| 863 | layers_[i]->ToProto(layer_param, write_diff); |
| 864 | } |
| 865 | } |
| 866 | |
| 867 | template <typename Dtype> |
| 868 | void Net<Dtype>::ToHDF5(const string& filename, bool write_diff) const { |