allenai / allenai/ask4help

Invalid Pretrained Checkpoints

Aperta
#1 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
22
Fork
1
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

It seems as though the pretrained checkpoints don't work with the current model. Running the given commands with the correct paths yields error

```
RuntimeError: Error(s) in loading state_dict for ResnetTensorObjectNavActorCritic:
size mismatch for goal_visual_encoder.resnet_compressor.0.weight: copying a param with shape torch.Size([128, 512, 1, 1]) from checkpoint, th
e shape in current model is torch.Size([128, 2048, 1, 1]).
size mismatch for actor.linear.weight: copying a param with shape torch.Size([4, 512]) from checkpoint, the shape in current model is torch.S
ize([6, 512]).
size mismatch for actor.linear.bias: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([6]
).
```

perhaps the checkpoints need to be updated? Please advise.

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.