Closed htwang14 closed 1 month ago
π Hello @htwang14, thank you for reaching out and for using YOLOv5 π!
To get started on your EMA-related query, I recommend reviewing our βοΈ Tutorials which provide comprehensive guides, including information on advanced training methods.
If your inquiry is a π Bug Report, please share a minimum reproducible example to help us diagnose the issue more effectively.
For custom training questions, such as applying EMA, provide additional details on your setup and confirm youβre following Tips for Best Training Results.
Ensure you have Python>=3.8.0 and all dependencies from requirements.txt installed, including PyTorch>=1.8. Begin with:
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
YOLOv5 can be run in these verified environments with all necessary dependencies:
Green badges indicate passing Continuous Integration (CI) tests for YOLOv5 training, validation, inference, export, and benchmarks across platforms.
Be sure to check out the latest YOLOv8 for enhanced capabilities in object detection, segmentation, and classification. Head to YOLOv8 Docs and start with:
pip install ultralytics
This is an automated response, but rest assured, an Ultralytics engineer will follow up to assist you shortly π.
The issue is more related with the ultralytics repo instead of yolov5. I've opened the same question there and closed this one. Thanks!
@htwang14 thank you for the update. If you need further assistance, feel free to reach out on the Ultralytics repo.
Search before asking
Question
Hi, I trained the yolo model using
model.train(data="coco8.yaml", epochs=100, imgsz=640)
and everything worked fine. I suppose the above code is not applying EMA, right? Could you please show me how to apply EMA during training using themodel.train
function?I've checked your code demo here. But it is not implemented with the
model.train
. Thank you in advance!Additional
No response