MyConfig.py
| 1 | from transformers import PretrainedConfig |
| 2 | from typing import List |
| 3 | |
| 4 | class RMBGConfig(PretrainedConfig): |
| 5 | model_type = "SegformerForSemanticSegmentation" |
| 6 | def __init__( |
| 7 | self, |
| 8 | in_ch=3, |
| 9 | out_ch=1, |
| 10 | **kwargs): |
| 11 | self.in_ch = in_ch |
| 12 | self.out_ch = out_ch |
| 13 | super().__init__(**kwargs) |
| 14 | |