xiuxiu__xiu的博客

原创 pytorch框架下的代码转换到jittor框架的过程中遇到的问题

最近正在尝试将pytorch框架下一个风格化网络（MCCNET)的代码转换到jittor框架下。在转换的过程中，大部分torch库中的函数都能在jittor库中找到作用相同的同名函数；小部分没能找到同名函数的也可以通过jittor库中的其他函数做到同样的效果。下面也主要是讲一下如何完成这几个空缺同名函数的实现方式，以及发现的一些因为jittor框架自身的特点带来的问题。我会在目录中将pytorch的函数名打出来，同时标注完成的操作，大家可以根据自己遇到的问题在目录中查询。MCCNET是中科院自动化所模式识