Frank Lee 6a3188167c set criterion as optional in colossalai initialize (#336)		2022-03-11 15:50:28 +08:00
..
__init__.py	set criterion as optional in colossalai initialize (#336)	2022-03-11 15:50:28 +08:00
_grad_scaler.py	Optimize pipeline schedule (#94)	2021-12-30 15:56:46 +08:00
torch_amp.py	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00