Open insafim opened 1 year ago
**trans = mt.Transformer() back = mb.Backbone(name='resnet50', train_backbone=False, return_interm_layers=False, dilation=False) model = DETR(backbone = back, transformer= trans, num_classes = 91, num_queries = 100)
model.load_state_dict(torch.load("detr-r50-e632da11.pth")) model.eval();**
I am getting the following error. _RuntimeError: Error(s) in loading state_dict for DETR:
**trans = mt.Transformer() back = mb.Backbone(name='resnet50', train_backbone=False, return_interm_layers=False, dilation=False) model = DETR(backbone = back, transformer= trans, num_classes = 91, num_queries = 100)
model.load_state_dict(torch.load("detr-r50-e632da11.pth"))
model.eval();**
I am getting the following error. _RuntimeError: Error(s) in loading state_dict for DETR: