torch.fx.experimental.optimization.reset_modules

torch.fx.experimental.optimization.reset_modules(nodes, modules, old_modules) [source]

Возвращает каждый модуль, изменённый с помощью modules_to_mkldnn, к исходному состоянию.