transformer_engine/pytorch/ops/sequential.py · 33a3d02f81c56e6f7b542c09bfa86657078d57fb · OpenDAS / TransformerEngine

"vscode:/vscode.git/clone" did not exist on "55f8b0a5def22ed6b85d3b91b726a7573d54313b"

[PyTorch] Update Sequential container to handle changes in module base class (#1028) · 33a3d02f

Tim Moon authored Jul 20, 2024



* Update sequential container constructor to handle modules in plain dicts
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Avoid initializing Sequential with dicts
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>

33a3d02f

sequential.py 5.63 KB

Replace sequential.py