Merge branch 'master' of github.com:Yidadaa/Pytorch-Video-Classification

README.md

@@ -11,15 +11,15 @@ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
 # CPU version
 conda install pytorch-cpu torchvision-cpu -c pytorch
 
-# 2. depencies
+# 2. pip dependencies
 pip install pandas scikit-learn tqdm opencv-python
 
 # 3. prepare datasets
 cd ./Pytorch-Video-Classification # go to the root dir of the code
 cp -r path/to/your/UCF ./data # copy UCF dataset to data dir
 cd ./data && python make_train_test.py # preprocess the dataset
 
-# 4. train your network
+# 4. train your network on UCF-101
 python train.py
 
 # (optional)5. restore from checkpoints
@@ -30,4 +30,4 @@ To know more about the usage of scripts, run the following commands:
 ```bash
 python train.py -h
 python make_train_test.py -h
-```
+```