Skip to content

Tags: ftgreat/apex

Tags

23.05

Toggle 23.05's commit message
source of apex in nvcr.io/nvidia/pytorch:23.05-py3

22.03

Toggle 22.03's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
[transformer] Update `build_model` function to support encoder&decode…

…r model (NVIDIA#1307)

* update build_model to support enc&dec model

* fix typo: cur_sargs -> cur_args

* enc&dec path: correctly update pre/post process