Browse Source

更新 'train_demo.py'

main
JiaChuan Shen 11 months ago
parent
commit
787a875cb8
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      train_demo.py

+ 3
- 3
train_demo.py View File

@@ -1,8 +1,8 @@
import os

os.system('python3 -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip')
os.system('python3 -m pip install -r /code/PaddleClas/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip')
# os.system('python3 -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip')
# os.system('python3 -m pip install -r /code/PaddleClas/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip')
os.system('python3 /code/PaddleClas/tools/train.py \
-c /code/PaddleClas/ppcls/configs/GeneralRecognitionV2/GeneralRecognitionV2_PPLCNetV2_base.yaml \
-c /code/PaddleClas/ppcls/configs/GeneralRecognitionV2/qizhi.yaml \
-o Arch.Backbone.pretrained=True \
-o Global.device=gpu')

Loading…
Cancel
Save