xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed yaml default

Browse Source
This commit is contained in:
nielseni6 2024-10-21 13:05:25 -04:00
parent 5953d3c9c6
commit 411157c18a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run_pgt_train.py
View File

@ -48,7 +48,7 @@ if __name__ == "__main__":
parser.add_argument('--device', type=str, default='0', help='CUDA device number') parser.add_argument('--device', type=str, default='0', help='CUDA device number')
parser.add_argument('--batch_size', type=int, default=64, help='Batch size for training') parser.add_argument('--batch_size', type=int, default=64, help='Batch size for training')
parser.add_argument('--epochs', type=int, default=100, help='Number of epochs for training') parser.add_argument('--epochs', type=int, default=100, help='Number of epochs for training')
parser.add_argument('--data_yaml', type=str, required=True, default='coco.yaml', help='Path to the data YAML file') parser.add_argument('--data_yaml', type=str, default='coco.yaml', help='Path to the data YAML file')
args = parser.parse_args() args = parser.parse_args()
# Set CUDA device (only needed for multi-gpu machines) # Set CUDA device (only needed for multi-gpu machines)