SaveNNMixin

class SaveNNMixin[source]

Bases: etna.core.mixins.SaveMixin

Implementation of AbstractSaveable torch related classes.

It saves object to the zip archive with 2 files:

  • metadata.json: contains library version and class name.

  • object.pt: object saved by torch.save.

Inherited-members

Methods

load(path)

Load an object.

save(path)

Save the object.