tests/components_to_test/simple_net.py · 75d221918af047e7323494907da3139a811791cd · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[Tensor] add 1d vocab loss (#918) · 75d22191
Ziyue Jiang authored May 07, 2022
```
* add 1d vocab loss

* polish
```
75d22191

simple_net.py 1.55 KB