first commit
This commit is contained in:
24
rtdetr_pytorch/tools/README.md
Normal file
24
rtdetr_pytorch/tools/README.md
Normal file
@@ -0,0 +1,24 @@
|
||||
|
||||
|
||||
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`
|
||||
|
||||
Reference in New Issue
Block a user