search

mmasana / FACIL

Framework for Analysis of Class-Incremental Learning with 12 state-of-the-art methods and 3 baselines.
https://arxiv.org/pdf/2010.15277.pdf
MIT License
512 stars 98 forks source link

Conversion of saved trained models (at checkpoints) #30

Closed andreadaou closed 1 year ago

andreadaou commented 1 year ago

Hello,

As described in the documentation, --save-models allows to save the trained models after each incremental step (task). Models are saved in a model.ckpt format without any other additional file (.meta file, .index file). How can I convert a saved model directly from .ckpt to .pb ?

Appreciate your help! Thank you in advance!

mmasana commented 1 year ago

Not sure how you can convert from one file type to another. But if you are using the disk logger on FACIL, then you can change the way that the models are saved by modifying this line here.

andreadaou commented 1 year ago

Okay, thank you! I will try to make some modifications.