search

bowenc0221 / panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)
Apache License 2.0
585 stars 117 forks source link

Train model Error: Unknown LR scheduler: WarmupPolyLR #90

Closed guoxuerui closed 3 years ago

guoxuerui commented 3 years ago

Hi Bowen, we occurred with error in LR scheduler setting. The problem is WarmupPolyLR is originally defined in DeepLab/build_solver but not involved in detectron2/solver/build. May I ask is there a way to use LR scheduler from the DeepLab separately? even if there's no such an option built yet in the solver of the whole detectron2 library?

bowenc0221 commented 3 years ago

Please check Panoptic-DeepLab example: you can import it like this and use it like this

guoxuerui commented 3 years ago

It works! thanks