Tim77777767 commited on
Commit
245b43a
·
1 Parent(s): e9318b0

README modified

Browse files
Files changed (2) hide show
  1. README.md +0 -1
  2. segformer-cityscapes-b5-bsm-hq.py +14 -0
README.md CHANGED
@@ -99,7 +99,6 @@ python3 -m segformer_plusplus.start_cityscape_benchmark --backbone [b1-b5] --hea
99
  ```
100
 
101
  Checkpoints can be downloaded via the provided links above:
102
- Example:
103
  ```shell
104
  wget -O [NameOfDownloadedFile] "URL of Model-Download"
105
  ```
 
99
  ```
100
 
101
  Checkpoints can be downloaded via the provided links above:
 
102
  ```shell
103
  wget -O [NameOfDownloadedFile] "URL of Model-Download"
104
  ```
segformer-cityscapes-b5-bsm-hq.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = ['../../../configs/segformer/segformer_mit-b5_8xb1-160k_cityscapes-1024x1024.py']
2
+
3
+ model = dict(
4
+ backbone=dict(
5
+ tome_cfg=[
6
+ dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
7
+ dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
8
+ dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4),
9
+ dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4)
10
+ ]
11
+ )
12
+ )
13
+
14
+ train_dataloader = dict(batch_size=4, num_workers=8)