Commit Graph

10 Commits (f7458d3ec7fa216b94dad7b9f10c6670a3252d46)

Author SHA1 Message Date
Jiarui Fang bc0e271e71
[buider] use builder() for cpu adam and fused optim in setup.py (#2187)
2 years ago
Frank Lee 81e0da7fa8
[setup] supported conda-installed torch (#2048)
2 years ago
Jiarui Fang 504419d261
[FAW] add cache manager for the cached embedding (#1419)
2 years ago
Frank Lee cf6d1c9284
[CLI] refactored the launch CLI and fixed bugs in multi-node launching (#844)
3 years ago
Frank Lee 01e9f834f5
[dependency] removed torchvision (#833)
3 years ago
Frank Lee 05d9ae5999
[cli] add missing requirement (#805)
3 years ago
Jiarui Fang 54229cd33e
[log] better logging display with rich (#426)
3 years ago
BoxiangW a2f1565672
Update GitHub action and pre-commit settings (#196)
3 years ago
Frank Lee 3defa32aee
Support TP-compatible Torch AMP and Update trainer API (#27)
3 years ago
zbian 404ecbdcc6 Migrated project
3 years ago