Question about F.relu(x + layer_func(x_shift)) in bn_inception.py

mit-han-lab / temporal-shift-module

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

https://arxiv.org/abs/1811.08383

MIT License

2.05k stars 418 forks source link

Question about F.relu(x + layer_func(x_shift)) in bn_inception.py #178

Open DuL1nk opened 3 years ago

DuL1nk commented 3 years ago

As stated in the paper, I got the same size x and x_shift. But the layer_func() changes the shape of x_shift which leads to an error when add it to x. Is there someone have the same problem?

russellllaputa commented 3 years ago

I've got the same problem in BNinception :(, but have no prboblem in mobilenetv2. Ask for solution!