if条件控制，尽量避免使用torch.jit.trace来转换代码，因为它不能处理变化条件，如果非要用trace的话，可以把if条件控制改成别的形式，比如：def f(x):
if x > 0:
return False
else:
return True
可以改成:
def f(x):
return x <= 0
jit不能转换第三方Python库中的函数，尽量所有代码都使用pytorch实现，如果速度不理想的话，可以参考github上的pytorch/extension-script项目，用C++实现需要的功能，然后注册成jit操作，最后转成torchscript。
如果要转Mobilenet，最好使用pytorch1.3以上，否则识别不出来其中的depth wise conv，转换出来的torchscript模型会比原模型大很多。
模型的forward函数中尽量不要包含中文注释。
如果代码中有if条件控制，尽量避免使用torch.jit.trace来转换代码，因为它不能处理变化条件，如果非要用...
注意事项 函数的默认参数如果不是tensor的话，需要指定类型 list中元素默认为tensor，如果不是，也要指...
一、使用展示物注意事项： 1展示物应该先藏好，直到使用时再拿出来。 2使用的展示物应该足够大，使最后一排的人都能够...
1：变量使用的注意事项 ①、变量定义后可以不赋值，使用时再赋值。不赋值不能使用。 ②、变量使用时有作用域的限制。 ...
使用时的注意事项: (CGRect)boundingRectWithSize:(CGSize)size optio...
【本章思想】 “我-信息”使用时的注意事项。 【本章思路】 介绍“我-信息”使用时的误区，提示家长需要多多练习。 ...
【本章思想】 “我-信息”使用时的注意事项。 【本章思路】 介绍“我-信息”使用时的误区，提示家长需要多多练习。 ...
由于机床事故发生频繁，为此小编整理了一份中走丝线切割机床的注意事项，大家在使用时要注意安全。 注意事项： 1、中走...
Dagger2 开发注意事项 1.component的inject方法接收父类型的参数，而调用时传入的参数是子类型...
1.类基本格式 class 类名{ } 2.使用时的注意事项: (1)类中的成员变量必须初始化(可以直接在定义变量...
本文标题：torch.jit使用时的注意事项（1）
本文链接：https://www.haomeiwen.com/subject/flwjgctx.html
网友评论