Skip to content

GitLab

Commit ae4d41f9 authored by Rayyyyy's avatar Rayyyyy
Browse files

Fix bug in train.sh

parent 35d29663
Hide 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."
\ No newline at end of file
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