Tags: lmw0320/apex
Tags
[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