File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ in **lib/network.py**.
12
12
- python 3.6
13
13
- pytorch 0.4.1
14
14
15
+
15
16
## Update Records
16
17
1 . Figure out how to implement the ** concatenation** type, and add the code to ** lib/** .
17
18
2 . Fix the bug in ** lib/non_local.py** (old version) when using multi-gpu. Someone shares the
@@ -35,7 +36,14 @@ to **Non-Local_pytorch_0.3.1/**.
35
36
2. Run **demo_MNIST.py** with one GPU or multi GPU.
36
37
```
37
38
CUDA_VISIBLE_DEVICES=0,1 python demo_MNIST.py
39
+
38
40
## Todo
39
- - Experiments on Kinetics dataset.
40
41
- Experiments on Charades dataset.
41
42
- Experiments on COCO dataset.
43
+
44
+
45
+ ## Related Repositories
46
+ 1. [Non-local ResNet-50 TSM](https://github.com/MIT-HAN-LAB/temporal-shift-module)
47
+ ([Paper](https://arxiv.org/abs/1811.08383)) on Kinetics dataset. They report that their model achieves a good performance
48
+ of **75.6% on Kinetics**, which is even higher than Non-local ResNet-50 I3D
49
+ ([Here](https://github.com/AlexHex7/Non-local_pytorch/issues/23)).
You can’t perform that action at this time.
0 commit comments