mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 05:24:22 +08:00
fixed yaml default
This commit is contained in:
parent
5953d3c9c6
commit
411157c18a
@ -48,7 +48,7 @@ if __name__ == "__main__":
|
||||
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('--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()
|
||||
|
||||
# Set CUDA device (only needed for multi-gpu machines)
|
||||
|
Loading…
x
Reference in New Issue
Block a user