Skip to content

Latest commit

 

History

History
22 lines (17 loc) · 799 Bytes

README.md

File metadata and controls

22 lines (17 loc) · 799 Bytes

non-local_pytorch

Statement

  • Only do the experiments on MNIST dataset so far.
  • You can find the non-local block in lib/.
  • If there is something wrong in my code, please contact me, thanks!

There are two version non-local.py and non-local-simple-version.py.

  • non-local.py contains the implementation of Gaussian, embedded Gaussian and dot product, which is mainly for learning.
  • non-local-simple-version.py only contains the implementation of embedded Gaussian.

Environment

  • python 3.6
  • pytorch 0.3.0

Todo

  • Experiments on Kinetics dataset.
  • Experiments on Charades dataset.
  • Experiments on COCO dataset.
  • Make sure how to do the Implementation of concatenation.