Wenwen Qu c1fe37d125 remove fp32 hook for norm		2023-09-25 10:56:55 +08:00
..
communication	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
context	feat(utils): add timeout warpper for key functions (#286)	2023-09-07 17:26:17 +08:00
scheduler	Merge remote-tracking branch 'origin/main' into develop	2023-09-08 10:19:54 +08:00
__init__.py	initial commit	2023-07-06 12:55:23 +08:00
engine.py	initial commit	2023-07-06 12:55:23 +08:00
gradient_handler.py	initial commit	2023-07-06 12:55:23 +08:00
naive_amp.py	remove fp32 hook for norm	2023-09-25 10:56:55 +08:00
trainer.py	feat: more tgs (#310)	2023-09-15 18:56:11 +08:00