Benchmark multi-model/multi-view models.
Note
This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the gallery for the big picture.
-
mmbench.workflow.smcvae.checkpoint(model, outfile, optimizer=None, scheduler=None, **kwargs)[source]ΒΆ Save the weights of a given model.
- Parameters
model : nn.Module
the model to be saved.
outfile : str
the destination file name.
optimizer : torch.optim.Optimizer
the optimizer.
scheduler : torch.optim.lr_scheduler, default None
the scheduler.
kwargs : dict
others parameters to be saved.
Follow us