set lm_head grad_mode=True - SemanticDiff

Commit

21 days ago

set lm_head grad_mode=True Signed-off-by: Xin He <xin3.he@intel.com>

References

#1971 - [AutoScheme] fix lm_head no_grad issue and support MOE model

Author

xin3he

xin3he

Parents

Loading