Hello,
I’m just wondering what the best practice is regarding storing models/model params together with e.g. classification metrics for that particular model.
Hello,
I’m just wondering what the best practice is regarding storing models/model params together with e.g. classification metrics for that particular model.
The recommended way is storing the state_dicts
as described in the serialization docs.
Often a dict
is stored with all state_dicts
and other attributes as shown in the ImageNet example.
Just what I needed, thanks!