#[WIP] Add Model Train on custom dataset SegFormer-tf Paper : https://arxiv.org/pdf/2105.15203 Model Architecture :