pytorch1.2,pytorch1.3如何使用1.1及以前生成的script(.pt)文件

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 */

发布了45 篇原创文章 · 获赞 21 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/xxradon/article/details/102968867