search

naseemap47 / YOLO-NAS

Train and Inference your custom YOLO-NAS model by Single Command Line
Apache License 2.0
98 stars 13 forks source link

training explain in detail info #13

Closed john09282922 closed 1 year ago

john09282922 commented 1 year ago

Thanks for nice works. I did well for training. But I'd like to change some hyper parameter, starting lr and weight decay, optimizer, etc. Could you send me some info to modify them?

thanks, john

naseemap47 commented 1 year ago

Hi @john09282922 , Thank you for using my repo. I will update these variables to training arguments ASAP. Thank you for your feedback. If you any other suggestion, please let me know.

naseemap47 commented 1 year ago

Hi @john09282922 Added lr and weight decay, optimizer, etc. to training arguments. I think this will help. Let me know your feedback. Thank you.