Skip to content

Tags: NVIDIA/apex

Tags

24.04.01

Toggle 24.04.01's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
Enhance Distributed Fused Adam (#1794)

23.08

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

23.07

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

23.06

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

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 (#1307)

* update build_model to support enc&dec model

* fix typo: cur_sargs -> cur_args

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