fastvideo.v1.training.checkpointing_utils
#
Module Contents#
Classes#
API#
- class fastvideo.v1.training.checkpointing_utils.ModelWrapper(model: torch.nn.Module)[source]#
- class fastvideo.v1.training.checkpointing_utils.OptimizerWrapper(model: torch.nn.Module, optimizer: torch.optim.Optimizer)[source]#
- class fastvideo.v1.training.checkpointing_utils.RandomStateWrapper(noise_generator: Optional[torch.Generator] = None)[source]#