open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
7.68k stars 2.53k forks source link

Trying to migrate 0.x version segmentor model to 1.2.2 version segmentor model. #3681

Open JKL-uxf opened 1 month ago

JKL-uxf commented 1 month ago

I was trying to implement Deepspeed or FSDP to a model (https://github.com/fudan-zvg/meta-prompts/blob/main/segmentation/README.md), but since I know 1.x.x version supports Deepspeed and FSDP now, so I am trying to change its source code to 1.2.2 version.

But I noticed that the model inherit BaseSegmentor has different parameters and receive different data, is there any simple way, or guide of this?