ver217 f8a7148dec [kernel] move all symlinks of kernel to colossalai._C (#1971)

2022-11-17 13:42:33 +08:00

..

grad_scaler

[NFC] polish amp.naive_amp.grad_scaler code style

2022-11-09 13:38:15 +08:00

__init__.py

[NFC] polish colossalai/amp/naive_amp/__init__.py code style (#1905)

2022-11-11 17:49:18 +08:00

_fp16_optimizer.py

[kernel] move all symlinks of kernel to colossalai._C (#1971)

2022-11-17 13:42:33 +08:00

_utils.py

[NFC] polish colossalai/amp/naive_amp/_utils.py code style (#1816)

2022-11-09 12:08:47 +08:00

naive_amp.py

[amp] included dict for type casting of model output (#1102)

2022-06-13 14:18:04 +08:00