Commit Graph

3 Commits (638a07a7f9b504e6c9781e9aa2a9b6c5e9dc49ed)

Author SHA1 Message Date
Frank Lee e7f3bed2d3
[booster] added the plugin base and torch ddp plugin (#3180)
2 years ago
Frank Lee 1ad3a636b1
[test] fixed torchrec model test (#3167)
2 years ago
Frank Lee ed19290560
[booster] implemented mixed precision class (#3151)
2 years ago