MnistModel.cpu

MnistModel.cpu() → T

Move all model parameters and buffers to the CPU.

Note

This method modifies the module in-place.

Returns:

Module: self

previous

MnistModel.compile

next

MnistModel.cuda