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

最新推荐文章于 2024-08-16 00:28:53 发布

丶Shining

最新推荐文章于 2024-08-16 00:28:53 发布

阅读量1.3k

点赞数

分类专栏： pytorch技巧

本文链接：https://blog.csdn.net/xxradon/article/details/102968867

版权

pytorch技巧专栏收录该内容

15 篇文章

订阅专栏

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