You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
1310967979@qq.com
51261fb4b3
|
7 months ago | |
|---|---|---|
| .. | ||
| data | 7 months ago | |
| det/ssd | 7 months ago | |
| finetune | 7 months ago | |
| README.md | 7 months ago | |
| train_parallel_with_func_example.py | 7 months ago | |
| train_with_func_example.py | 7 months ago | |
README.md
This folder contains examples for various tasks, which users can run easily.
Single process with model training and evaluation
python examples/train_with_func_example.py
This example shows how to train and evaluate a model on your own dataset.
Multiprocess with model training and evaluation
export CUDA_VISIBLE_DEVICES=0,1,2,3 # suppose there are 4 GPUs
mpirun --allow-run-as-root -n 4 python examples/train_parallel_with_func_example.py
This example shows how to train and evaluate a mode with multiprocess on your own dataset on GPU.