ModelScope报错怎么办?

ModelScope报错怎么办?

2023-08-23 16:30:47,857 - modelscope - INFO - Text logs will be saved to tmp_dir
Traceback (most recent call last):
  File "D:/pythonWarkSpace/modelscope_test/mgeo_train.py", line 88, in <module>
    finetune(
  File "D:/pythonWarkSpace/modelscope_test/mgeo_train.py", line 27, in finetune
    trainer.train()
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\modelscope\trainers\trainer.py", line 693, in train
    self.train_loop(self.train_dataloader)
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\modelscope\trainers\trainer.py", line 1215, in train_loop
    self.train_step(self.model, data_batch)
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\modelscope\trainers\trainer.py", line 860, in train_step
    train_outputs = model.forward(**inputs)
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\modelscope\models\multi_modal\mgeo\token_classification.py", line 214, in forward
    loss = loss_fct(active_logits, active_labels)
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\torch\nn\modules\loss.py", line 1174, in forward
    return F.cross_entropy(input, target, weight=self.weight,
  File "E:\Anaconda3\envs\modelscope\lib\site-packages\torch\nn\functional.py", line 3029, in cross_entropy
    return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: expected scalar type Long but found Int

展开
收起
青城山下庄文杰 2023-08-25 15:47:39 161 分享 版权
1 条回答
写回答
取消 提交回答
  • 您看一下torch版本,linux上和您的报错不一致。-此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-08-29 15:40:12
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理