Not sure how to save model in h5 format

I am using following as reference. It appears there are two main ways to save the model:

  • save dictionary state: recommended way.
  • save entire model: not recommended as path is saved and can break if it is opened somewhere else.

TF can conveniently store in h5 format which I can dump quickly using h5 (hdf5) format, but this is not possible with Pytorch?

At least according to this, hdf5 format is not great as it accumulates over time. Is there a better format these days?

Pytorch usually use .pth or .pt

those are the formats that are mostly used in pytorch