Angel-ML / PyTorch-On-Angel

PyTorch On Angel, arming PyTorch with a powerful Parameter Server, which enable PyTorch to train very big models.
164 stars 51 forks source link

modify gen_pt_model.sh for docker env Ye Huanjie #111

Open YeHuanjie opened 2 years ago

YeHuanjie commented 2 years ago

modify gen_pt_model.sh