Pytorch JIT 使用经验总结

注意

1. 使用CPU模式进行Trace或者jit.script正常，但是使用GPU模式确报错。

所有的torch.operator，需要传dim的。如

torch.cat(tensors, dim=0, out=None) → Tensor

传参数必要要注意，dim绝对不能** =-1 **，因为使用Trace的时候，CPU虽然可能一切正常，但是使用GPU进行Trace的时候会报错。一般会报如下错误：

torch.jit.TracingCheckError: Tracing failed sanity checks!
Encountered an exception while running the trace with test inputs.
Exception:
	vector::_M_range_check: __n (which is 18446744073709551615) >= this->size() (which is 2)