aai-institute / ProtoTree

Fork of ProtoTrees: Neural Prototype Trees for Interpretable Fine-grained Image Recognition, published at CVPR2021
Other
3 stars 1 forks source link

Make new models saveable #36

Open Clifford-appliedAI-GmbH opened 1 year ago

Clifford-appliedAI-GmbH commented 1 year ago

The new Lightning models cannot currently be saved. The problem was that the old model saving code didn't work properly (due to various critical fields not being saved correctly), so it was removed for now.

This seems like it wouldn't be too difficult to reintroduce, but we should confirm this.