ab3b01867c update
PCL-陶恒韬
taoht
- Joined on Apr 22, 2021
- Organization
-
-
-
32
-
2
-
-
15
-
22
-
Loading Heatmap…
taoht commented on issue PCL-Platform.Inte.../mPanGu-Alpha-53#11
GPU方式训练保存的8份checkpoint如何合并然后进行模型推理1、分布式加载8片模型参数 pangu_alpha = PanguAlphaModel(config) eval_net = EvalNet(pangu_alpha, pad_token=args_opt.padding_id) eval_net.set_train(False) model_predict = Model(eval_net) load_distributed_checkpoint(eval_net, ckpt_file_list, predict_layout) 2、整合保存成单个模型文件 save_checkpoint(pangu_alpha, "/path/to/mPanGu_integrated.ckpt", integrated_save=True)
1 month ago
taoht commented on issue PCL-Platform.Inte.../mPanGu-Alpha-53#7
使用lo_mn的模型推理速度较慢推理时需配置参数： use_past=true use_pynative_op=1
2 months ago
taoht upload dataset mPanGu_Alpha-53_mn-lo.ckpt
6 months ago