pytorch自从1.2版本以后,主要的数据结构上面的变化就是增加了bool变量,所以以前的uint8输入需要更改一下。
不解压,使用压缩工具软件,如rar,打开.pt文件,如图。
找到里面的code/xray.py文件,这里面记录的就是我们的静态图,所有的参数都是固定写好的。
将dtype=0 的改为dtype=11
下面是c10/core/ScalarType.h中定义的顺序
#define AT_FORALL_SCALAR_TYPES_WITH_COMPLEX_AND_QINTS(_) \
_(uint8_t, Byte) /* 0 */ \
_(int8_t, Char) /* 1 */ \
_(int16_t, Short) /* 2 */ \
_(int, Int) /* 3 */ \
_(int64_t, Long) /* 4 */ \
_(at::Half, Half) /* 5 */ \
_(float, Float) /* 6 */ \
_(double, Double) /* 7 */ \
_(at::ComplexHalf, ComplexHalf) /* 8 */ \
_(std::complex<float>, ComplexFloat) /* 9 */ \
_(std::complex<double>, ComplexDouble) /* 10 */ \
_(bool, Bool) /* 11 */ \
_(c10::qint8, QInt8) /* 12 */ \
_(c10::quint8, QUInt8) /* 13 */ \
_(c10::qint32, QInt32) /* 14 */ \
_(at::BFloat16, BFloat16) /* 15 */