facebookresearch / facebookresearch/SlowFast

Network weights head.projection.weight not loaded. Network weights head.projection.bias not loaded.

Open
#333 2 comments 0 reactions 0 assignees View on GitHub
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:

![image](https://user-images.githubusercontent.com/43233772/99745356-6e039180-2b14-11eb-82d4-3cc9ac299b27.png)

They really have differences at the last of text. How to fix this problem ?

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.