torch.jit使用时的注意事项(1)

  1. 如果代码中有if条件控制,尽量避免使用torch.jit.trace来转换代码,因为它不能处理变化条件,如果非要用trace的话,可以把if条件控制改成别的形式,比如:
def f(x):
  if x > 0:
    return False
  else:
    return True

可以改成:

def f(x):
  return x <= 0
  1. jit不能转换第三方Python库中的函数,尽量所有代码都使用pytorch实现,如果速度不理想的话,可以参考github上的pytorch/extension-script项目,用C++实现需要的功能,然后注册成jit操作,最后转成torchscript

  2. 如果要转Mobilenet,最好使用pytorch1.3以上,否则识别不出来其中的depth wise conv,转换出来的torchscript模型会比原模型大很多。

  3. 模型的forward函数中尽量不要包含中文注释。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 本次笔记使用的材料是2018版的斯坦福计算机课程Lecture8的ppt,相较于2017版改动了一些内容,总体是比...
    HRain阅读 980评论 0 1
  • 人生苦短我用 Python 注:最后附电子书地址 一、Pythonic Thinking 第1条: 确认自己所用的...
    molscar阅读 2,061评论 0 3
  • 10 夏杨挣扎想要爬起来,却被方林紧紧的按在怀里道:“很想我早点休息吗?” 方林的声音很轻很低,快低到了夏杨的心...
    星如雨雨雨阅读 248评论 0 0
  • 需求第二定律 作者:薛兆丰 需求第二定律有很多版本。需求第二定律是说:需求对价格的弹性,和价格变化之后流逝的时间长...
    RockyReading阅读 1,213评论 0 0
  • 学完前面的语法知识,应该说Swift的语法已经基本讲完,Swift是一门正在走向成熟的年轻语言,每次发布新版本都会...
    YungFan阅读 495评论 0 2