ColossalAI/colossalai/nn/_ops/loss.py at e1f97fd2b850f5819afd448b5489e18e50fd91ef - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-29 21:49:54 +00:00

Files

HELSON 0453776def [tensor] fix a assertion in colo_tensor cross_entropy (#1232 )

2022-07-08 11:18:00 +08:00

2.6 KiB

Raw Blame History

View Raw