25 lines
737 B
Markdown
25 lines
737 B
Markdown
|
|
|
|
Train/test script examples
|
|
- `CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --nproc_per_node=4 --master-port=8989 tools/train.py -c path/to/config &> train.log 2>&1 &`
|
|
- `-r path/to/checkpoint`
|
|
- `--amp`
|
|
- `--test-only`
|
|
|
|
|
|
Tuning script examples
|
|
- `torchrun --master_port=8844 --nproc_per_node=4 tools/train.py -c configs/rtdetr/rtdetr_r18vd_6x_coco.yml -t https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r18vd_5x_coco_objects365_from_paddle.pth`
|
|
|
|
|
|
Export script examples
|
|
- `python tools/export_onnx.py -c path/to/config -r path/to/checkpoint --check`
|
|
|
|
|
|
GPU do not release memory
|
|
- `ps aux | grep "tools/train.py" | awk '{print $2}' | xargs kill -9`
|
|
|
|
|
|
Save all logs
|
|
- Appending `&> train.log 2>&1 &` or `&> train.log 2>&1`
|
|
|