MnistModel.half

MnistModel.half() → T

Casts all floating point parameters and buffers to half datatype.

Note

This method modifies the module in-place.

Returns:

Module: self

previous

MnistModel.get_submodule

next

MnistModel.load_state_dict