MnistModel.to_empty

MnistModel.to_empty(*, device: Union[str, device]) → T

Moves the parameters and buffers to the specified device without copying storage.

Args:

device (torch.device): The desired device of the parameters

and buffers in this module.

Returns:

Module: self

previous

MnistModel.to

next

MnistModel.train