RuntimeError: weight tensor should be defined either for all 9 classes or no classes but got weight tensor of shape: [1, 9]

JacobYuan7 / DIN-Group-Activity-Recognition-Benchmark

[ICCV 2021] A new codebase containing various methods for Group Activity Recognition. Paper title: Spatio-Temporal Dynamic Inference Network for Group Activity Recognition.

MIT License

52 stars 15 forks source link

RuntimeError: weight tensor should be defined either for all 9 classes or no classes but got weight tensor of shape: [1, 9] #16

Closed Xiyu-AI closed 1 year ago

Xiyu-AI commented 1 year ago

Hello, I run the train_volleyball_stage1.py and I met this question. Could you please help me to solve this question? Thanks!