junyongyou / triq

TRIQ implementation
MIT License
134 stars 23 forks source link

Save model config data #26

Closed codingzebra333 closed 1 year ago

codingzebra333 commented 1 year ago

Hey, I wanted to ask which would be the best way to save model config data? After training it only saves weights and not the model config data itself. I tried changing it in callbacks.py "save_weights_only=False", but it did not work. Are there any other ways how to deal with this? Thank you in advance!