Fix bug in train.sh

train.sh

@@ -8,6 +8,6 @@ export MIOPEN_FIND_MODE=4
 
 echo "Training start ..."
 # 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."