How to save model in mnist.cpp?

[engineer1109]

How to save the model in cpp api mnist.cpp?
model.save and torch::save(model,"mnisttrain.pkl")
All error

[gaohaihui]

parser.add_argument('--save-model', action='store_true', default=False,
help='For Saving the current Model')
default rewrite True in parameter

[engineer1109]

@gaohaihui I mean the minst.cpp ,c++ api ,not python

[xiaoLiuxiaoLiuxiaoLiu]

You can save all layers weights and bias at a vector, and torch::save and torch::load will be useful in this situation. Any easier implement ? Plz comment.

[soumith]

cc: @yf225

[xiaoLiuxiaoLiuxiaoLiu]

Here is a very stupid but useful implement of model_load and model_save.
In model_load, Load a std::vectortorch::Tensor and save every weights and bias in a "inter_layer.pt" file. And then, with the code of torch::load(model.parameters()[i], "inter_layer.pt"), layers parameters can be loaded.
In contrary, with the code of { std::vector torch::Tensor a; a.push_back(model.parameters()[i])}, layers parameters can be saved as Vector File.
Unfortunately, with my test, a complex model with many structures cannot be save and load directly. Plz comment if any easier implement exists.