Saving model about couta

[Eric-Simon-IA]

Well I managed to make something work now, by saving state_dict, dim and c, but not sure if it's the best way to do that

[xuhongzuo]

Hi @Eric-Simon-IA, I just added model save and load to couta_algo.py. You can now assign save_model_path and load_model_path when initializing COUTA to realize saving the trained network or loading of the fitted network. I also update a section in the Readme.md.

Thanks again for raising this issue and sorry for the late update.
I will close this issue, and if you have any further problems related to this problem, please feel free to reopen it.

Cheers! Hope you have a wonderful week!

[Eric-Simon-IA]

Thank you, will look at it when I manage to get some free time.
Hope you have a nice week as well !

[xuhongzuo]

Thanks for raising this issue. I will add model saving and loading into the notebook file.
Yes, saving state_dict of the network, the center c, and the dimensionality dim seems to be a good solution.