Created by: littletomatodonkey
Add supprt for windows and CPU.
-
set the environment
- for windows:
SET PYTHONPATH=$PWD $PYTHONPATH
- for linux:
export PYTHONPATH=$PWD:$PYTHONPATH
- for windows:
-
train
python tools/train_multi_platform.py -c configs/ResNet/ResNet50.yaml
- eval
python -u tools/eval_multi_platform.py -c configs/ResNet/ResNet50.yaml -o pretrained_model=./output/ResNet50/1/ppcls -o VALID.batch_size=2
- If you want to use cpu to train, you should use
tools/train_multi_platform.py
, and setuse_gpu
asFalse
explicitly in the config file(*.yaml). The following is an example.
mode: 'train'
ARCHITECTURE:
name: 'ResNet50'
pretrained_model: ""
model_save_dir: "./output/"
classes_num: 1000
use_gpu: False
..
..
..