Skip to content

GitLab

"magic_pdf/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "0a1fb1e431756f0bc4b40a64d6fa38a26ad21e4a"
Commit ae4d41f9 authored by Rayyyyy's avatar Rayyyyy
Browse files

Fix bug in train.sh

parent 35d29663
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
train.sh
View file @ ae4d41f9
...@@ -8,6 +8,6 @@ export MIOPEN_FIND_MODE=4 ...@@ -8,6 +8,6 @@ export MIOPEN_FIND_MODE=4
echo "Training start ..." echo "Training start ..."
# coco_path是训练数据集地址，数据是coco format # coco_path是训练数据集地址，数据是coco format
torchrun --nproc_per_node=4 --use_env main.py --coco_path /path/to/coco torchrun --nproc_per_node=4 main.py --coco_path /path/to/coco
echo "Training finished." echo "Training finished."
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment