# Train 1 gpu
# python train.py --img 800 --batch 16 --epochs 100 --data data/kwater.yaml --weights yolov5s.pt

# Train multi gpu
python -m torch.distributed.run --nproc_per_node 2 train.py --img 800 --batch 32 --epochs 30 --data data/kwater.yaml --cfg yolov5m.yaml --weight '' --device 0,1

# python -m torch.distributed.run --nproc_per_node 2 train.py --img 256 --batch 32 --epochs 100 --data data/kwater-custom.yaml --weights yolov5s.pt --device 0,1

# Fine-tune multi gpu
# python -m torch.distributed.run --nproc_per_node 2 train.py --img 800 --batch 32 --epochs 20 --data data/construction_worker.yaml --weights '/root/DataSet/HD/yolov5/runs/train/exp10/weights/best.pt' --device 0,1

# Val
# python val.py --weights /root/DataSet/HD/yolov5/runs/train/finetunning/weights/last.pt --data data/crowd_human.yaml --img 800

# python detect.py --weights /root/DataSet/HD/yolov5/runs/train/exp10/weights/last.pt --source data/images