!1181 Fix Prediction Shape Check For PanGuAlpha
Merge pull request !1181 from huangxinjing/fix_inference_error
Showing
- official/nlp/pangu_alpha/README.md 9 additions, 6 deletionsofficial/nlp/pangu_alpha/README.md
- official/nlp/pangu_alpha/predict.py 1 addition, 0 deletionsofficial/nlp/pangu_alpha/predict.py
- official/nlp/pangu_alpha/scripts/run_standalone_export.sh 3 additions, 3 deletionsofficial/nlp/pangu_alpha/scripts/run_standalone_export.sh
- official/nlp/pangu_alpha/src/pangu_alpha.py 5 additions, 4 deletionsofficial/nlp/pangu_alpha/src/pangu_alpha.py
- official/nlp/pangu_alpha/src/pangu_alpha_config.py 4 additions, 2 deletionsofficial/nlp/pangu_alpha/src/pangu_alpha_config.py
Please register or sign in to comment