Shortcuts

torchtnt.utils.stateful.Stateful

class torchtnt.utils.stateful.Stateful(*args, **kwargs)

Defines the interface for checkpoint saving and loading.

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)
load_state_dict(state_dict)
state_dict()

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources