Frank Lee c35fdbfe5d set criterion as optional in colossalai initialize (#336)

2022-03-09 11:51:22 +08:00

..

__init__.py

set criterion as optional in colossalai initialize (#336)

2022-03-09 11:51:22 +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