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