We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ec70f2 commit 052ea17Copy full SHA for 052ea17
video/README.md
@@ -9,12 +9,12 @@ Download the pretrained models on classification in [here](https://drive.google.
9
10
On UCF101:
11
```
12
-CUDA_VISIBLE_DEVICES=0,1,2,3 python train_ucf101.py
+python train_ucf101.py --batch-size 32 --gpus 0,1
13
14
15
On HMDB51:
16
17
-CUDA_VISIBLE_DEVICES=0,1,2,3 python train_hmdb51.py
+python train_hmdb51.py --batch-size 32 --gpus 0,1
18
19
20
### Test
0 commit comments