Skip to content

some question about model code #37

@mengfeiwu-ai

Description

@mengfeiwu-ai

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions