Update README.md

tejas1904 · Mar 27, 2023 · 9e05b7c · 9e05b7c
1 parent 44aa19b
commit 9e05b7c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,8 @@
 #[WIP]
 
 - [x] Add Model
-- [ ] Train on custom dataset
+- [ ] Train on ADE20k dataset
+- [ ] Train on cityscapes
 
 # SegFormer-tf
 
@@ -21,4 +22,4 @@ Paper : https://arxiv.org/pdf/2105.15203
 
 <p align="center">
   <img src="https://github.com/IMvision12/SegFormer-tf/blob/main/images/mit.png" title="graph">
-</p>
+</p>