HELSON a30e2b4c24 [zero] adapt for no-leaf module in zero (#535) ...

only process module's own parameters in Zero context

add zero hooks for all modules that contrain parameters

gather parameters only belonging to module itself

2022-03-28 17:42:18 +08:00

..

utils

[unit test] Refactored test cases with component func (#339)

2022-03-11 15:50:28 +08:00

__init__.py

[zero] adapt for no-leaf module in zero (#535)

2022-03-28 17:42:18 +08:00

bert.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

nested_model.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

no_leaf_module.py

[zero] adapt for no-leaf module in zero (#535)

2022-03-28 17:42:18 +08:00

registry.py

[test] refactored testing components (#324)

2022-03-11 15:50:28 +08:00

repeated_computed_layer.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

resnet.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00