Bering.models.BaselineMLP.type

BaselineMLP.type(dst_type)

Casts all parameters and buffers to dst_type. :rtype: TypeVar(T, bound= Module)

Note

This method modifies the module in-place.

Args:

dst_type (type or string): the desired type

Returns:

Module: self