Skip to content

Commit 30acf8a

Browse files
committed
add Info about TSM Repo.
1 parent 4cdb732 commit 30acf8a

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

README.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ in **lib/network.py**.
1212
- python 3.6
1313
- pytorch 0.4.1
1414

15+
1516
## Update Records
1617
1. Figure out how to implement the **concatenation** type, and add the code to **lib/**.
1718
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/**.
3536
2. Run **demo_MNIST.py** with one GPU or multi GPU.
3637
```
3738
CUDA_VISIBLE_DEVICES=0,1 python demo_MNIST.py
39+
3840
## Todo
39-
- Experiments on Kinetics dataset.
4041
- Experiments on Charades dataset.
4142
- 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)).

0 commit comments

Comments
 (0)