type

UNet.type(dst_type: Union[torch.dtype, str]) → torch.nn.modules.module.T

Casts all parameters and buffers to dst_type.

Note

This method modifies the module in-place.

Parameters

dst_type (type or string) – the desired type

Returns

self

Return type

Module