facebookresearch / facebookresearch/SlowFast
Network weights head.projection.weight not loaded. Network weights head.projection.bias not loaded.
- Dominant language
- Python
- Stars
- 7.4k
- Forks
- 1.3k
- PR merge metrics
- No merged PRs in 30d
Description
I trained on my custom dataset with 6 action classes, one of the video is like [this](https://streamja.com/rLJ6e), it belongs to the ''fall_down" action class.
After training, i want to run the demo with trained checkpoint, but the output video doesn't recognize any action, looking at the log, it says
```
[11/20 08:58:46][INFO] checkpoint.py: 336: Network weights head.projection.weight not loaded.
[11/20 08:58:46][INFO] checkpoint.py: 336: Network weights head.projection.bias not loaded.
```
I print out `model_dict.keys()` and `pre_train_dict_match.keys()` at about [here](https://github.com/facebookresearch/SlowFast/blob/fd41618191d3c21c1ca21a61369ce9917646cf9c/slowfast/utils/checkpoint.py#L323), then i got the log text, and compare them using [diffchecker](https://www.diffchecker.com/diff) tool to check their differnces, the result shows as below:

They really have differences at the last of text. How to fix this problem ?
Contributor guide
Assessment
This issue has not been assessed yet.