Best practice on saving model params together with metrics


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.

1 Like

Just what I needed, thanks!