作者您好,我想用你们的代码在我的数据集上进行训练和测试,但是我遇到了一些问题:
您在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作为字典操作,也报了这个错误。
作者您好,我想用你们的代码在我的数据集上进行训练和测试,但是我遇到了一些问题:
您在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作为字典操作,也报了这个错误。