ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

History

HELSON 42ab36b762 [tensor] add unitest for colo_tensor 1DTP cross_entropy (#1230 )		2022-07-07 19:17:23 +08:00
..
__init__.py	…
loss_1d.py	[tensor] add unitest for colo_tensor 1DTP cross_entropy (#1230 )	2022-07-07 19:17:23 +08:00
loss_2d.py	…
loss_2p5d.py	…
loss_3d.py	…
loss_moe.py	…