You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
作者您好,我想用你们的代码在我的数据集上进行训练和测试,但是我遇到了一些问题:
您在PluginMoodel中定义您的主要模型,在文件pim_module.py第330行有以下代码:
# get hidden feartues size
rand_in = torch.randn(1, 3, img_size, img_size)
outs = self.backbone(rand_in)
这个outs是一个tensor吧,但是后面用它作为字典进行查询,所以报了以下错误:
IndexError: tensors used as indices must be long, int, byte or bool tensors
另外FPN中也将outs作为字典操作,也报了这个错误。
The text was updated successfully, but these errors were encountered:
作者您好,我想用你们的代码在我的数据集上进行训练和测试,但是我遇到了一些问题:
您在PluginMoodel中定义您的主要模型,在文件pim_module.py第330行有以下代码:
# get hidden feartues size
rand_in = torch.randn(1, 3, img_size, img_size)
outs = self.backbone(rand_in)
这个outs是一个tensor吧,但是后面用它作为字典进行查询,所以报了以下错误:
IndexError: tensors used as indices must be long, int, byte or bool tensors
另外FPN中也将outs作为字典操作,也报了这个错误。
The text was updated successfully, but these errors were encountered: