专栏算法工具链QAT量化训练模型evaluate数据传入

QAT量化训练模型evaluate数据传入

已解决
默认328322024-08-27
100
4

J5 J5_OE_1.1.74

之前我们使用Batch对所有数据进行封装并传入模型,出现dtype报错如下:

按照论坛回复,calibration模型仅接受单个tensor类型或dict字典类型,我们对数据类型进行了修改,整理成如下格式:

每个变量都以字典格式进行输入,但是仍然出现报错:

请问对于多输入的网络,你们的数据传入是如何进行的?

算法工具链
技术深度解析征程5
评论2
0/1000
  • 遥看瀑布挂前川
    Lv.2

    得是一个迭代器，torch.dataloader

    2024-08-28
    0
    2
    • 默认31869回复遥看瀑布挂前川:

      目前我们使用的是torch_geometric的dataloader,将数据(Data)包装成Batch无法进入你们这边的calib_model与qat量化;训练的model,更换了包装形式以后,以字典的形式依然无法进入

      2024-08-29
      0
    • Huanghui回复默认31869:
      你好，上面已经提示的比较清楚了，传入的数据应该是个torch.dataloader对象或子类，torch.utils.data.DataLoader 和 torch_geometric.loader.DataLoader 是两个不同的加载器。
      2024-09-03
      0
  • Huanghui
    Lv.5

    客户您好，长时间未收到你的答复，相信问题已解。如对此尚存疑问欢迎新帖讨论，感谢您的参与！

    2025-07-09
    0
    0