昇腾社区首页
中文
注册
开发者
下载

模型选取

建议用户在选取迁移模型时,尽可能选取权威PyTorch模型实现仓库,包括但不限于PyTorch(imagenet/vision等)、Meta Research(Detectron/detectron2等)、open-mmlab(MMDetection/mmpose等)。

对于大模型,使用较为广泛的资源仓库是HuggingFaceMegatron-LMLlama-Factory等仓库,可以在其中选取目标模型。

评估选取模型迁移的可行性:

  • 保证选取的模型能在三方平台(如GPU)可成功运行。
  • 从权威网站或数据平台获取选定模型的精度和性能基线,或在三方平台实测精度和性能基线。