allenai / allenai/ask4help

Invalid Pretrained Checkpoints

Open
#1 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
22
Forks
1
PR merge metrics
No merged PRs in 30d

Description

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.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.