search

dbolya / yolact

A simple, fully convolutional model for real-time instance segmentation.
MIT License
5k stars 1.33k forks source link

Apply different lr scheduler of pytorch for yolact++ #736

Open Elizbellou opened 2 years ago

Elizbellou commented 2 years ago

Hello, I would like to change the default lr scheduler for yolact using one of the pytorch lr schedulers in an attempt to improve mAP (I use September's update yolact++ with custom dataset). So, in which file should I make changes and which lines? Thank u in advance for your help

JayParikh20 commented 1 year ago

@Elizbellou did you find any solution for this?

Elizbellou commented 1 year ago

@Elizbellou did you find any solution for this?

Sorry, for the delay.aybe you've already found a solution. I didn't exactly used a lr scheduler algorithm, however by just changing lr and lr steps according to max iterations in the config file, it raised mAP about 2-4%. By default it multiplies lr0 with 0.1 at every lr step, if I remember well.. So, to your question if you find the file where this math equation is defined, u can change that with your own schedule. If changes to other files are required I guess it will bring up an error so you go along and fix it.