安装
pip install mindtorch
mindtorch 用于帮助迁移torch模型到mindspore
大部分都可以直接把mindtorch的torch搞成torch,就和以前的代码一致,注意下面
只有静态图有点点差异
step也有差异
自定义优化器就麻烦了。
pyttorch还是牛啊
并行计算还是用的ms的
还支持cpu的并行,还挺好
注意的东西不少啊,为什么依赖pytorch呢
我建议还是要做一些实际的差异点例子,我指的不是实战迁移,而是把某段迁移代码的智能ai助手搞出来,当用户想迁移的时候,就智能写出来代码,而不是人为的去注意差异点。
我这个点子是不是很棒?