heat.nn
This is the heat.nn submodule.
It contains data parallel specific nn modules. It also includes all of the modules in the torch.nn namespace
Submodules
Package Contents
- __getattr__(name)
When a function is called for the heat.nn module it will attempt to run the heat nn module with that name, then, if there is no such heat nn module, it will attempt to get the torch nn module of that name.