Hyper-parameter optimization of YOLOv7 model

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.26k stars 4.18k forks source link

Hyper-parameter optimization of YOLOv7 model #2017

Open elon1992 opened 5 months ago

elon1992 commented 5 months ago

Hello @WongKinYiu ,

I am trying to do hyper-parameter optimization using Optuna, so anyone suggest me the changes that i need to do in Train_aux.py in order to finetune the model using Bayesian Optimization method. The help would be really appreciable! Thank you!