Non-local_pytorch
Non-local_pytorch copied to clipboard
Implementation of Non-local Block.
Awesome library, I was able to convert it to train against my own dataset after making some modifications. Are there any plans to include (or do you have something personal...
Hi~ I use the code but some problems happened , my torch version is 1.1.0, is it must to use the 0.4.0 torch?
When setting subsample to True,why self.theta don't add a maxpool layer?
If you test Softmax instead scaling by 1/N
non-local doesn't work in my experiments (I3D,UCF101),Is this normal?
Hi, I have trained a Non-local ResNet-50 TSM model on Kinetics dataset using your code here: https://github.com/MIT-HAN-LAB/temporal-shift-module (I have stated the implementation is from your GitHub repo in my code)....
Thanks for providing us this wonderful repository. But I have a question about how to initialize the parameters in g、theta and phi when training。Thanks!
Hi!Is this module computationally heavy? When I add this module to my network, there are always CUDA out of memory errors. However, when I remove this module, it will work...
代码深入浅出,清晰易懂,点赞(Thanks for author for opening the source code. It's a great job)
感谢你很棒的工作,如果使用更大的数据集?例如CIFAR100?能提供可以跑多个不同数据集的训练脚本吗?谢谢