Non-local_pytorch icon indicating copy to clipboard operation
Non-local_pytorch copied to clipboard

Implementation of Non-local Block.

Results 32 Non-local_pytorch issues
Sort by recently updated
recently updated
newest added

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?能提供可以跑多个不同数据集的训练脚本吗?谢谢