OutHeadFC.cpu
- OutHeadFC.cpu() T[source]
Move all model parameters and buffers to the CPU.
Note
This method modifies the module in-place.
- Returns:
Module: self
Move all model parameters and buffers to the CPU.
Note
This method modifies the module in-place.
Module: self