utae-paps icon indicating copy to clipboard operation
utae-paps copied to clipboard

Hyperparameters with UTAE model

Open PedroConrado opened this issue 10 months ago • 1 comments

Currently trying to reproduce the results of the segmentation experiments on the PASTIS benchmark using the UTAE model. Were the default script value used (learning rate of 0.001 and batch size of 4), or different hyperparameters were applied?

PedroConrado avatar Feb 15 '25 19:02 PedroConrado

Hi @PedroConrado In the paper, the semantic segmentation experiments use a fixed learning rate of 0.001 for 100 epochs. For the panoptic segmentation experiments, we start with a higher learning rate of 0.01 for 50 epochs, and decrease it to 0.001 for the last 50 epochs (see appendix of the paper).

VSainteuf avatar Feb 26 '25 14:02 VSainteuf