Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.optimization.remove_dropout#

torch.fx.experimental.optimization.remove_dropout(model)[source]#

Removes all dropout layers from the module.

Return type:

Module

On this page
Show Source
PyTorch Libraries