MnistModel.type

MnistModel.type(dst_type: Union[dtype, str]) → T

Casts all parameters and buffers to dst_type.

Note

This method modifies the module in-place.

Args:

dst_type (type or string): the desired type

Returns:

Module: self

previous

MnistModel.train

next

MnistModel.xpu