Skip to content

Commit 73650cb

Browse files
committed
add unetformer code
1 parent ba53673 commit 73650cb

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

docs/zh/examples/unetformer.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,9 @@ examples/unetformer/geoseg/losses/joint_loss.py:23:40
171171
--8<--
172172
```
173173
#### 3.4.2 UNetFormerLoss
174-
``` py linenums="92"
174+
``` py linenums="93"
175175
--8<--
176-
examples/unetformer/geoseg/losses/useful_loss.py:92:113
176+
examples/unetformer/geoseg/losses/useful_loss.py:93:114
177177
--8<--
178178
```
179179

@@ -189,9 +189,9 @@ examples/unetformer/config/vaihingen/unetformer.py:72:83
189189

190190
### 3.6 模型训练
191191

192-
``` py linenums="238"
192+
``` py linenums="236"
193193
--8<--
194-
examples/unetformer/train_supervision.py:238:302
194+
examples/unetformer/train_supervision.py:236:300
195195
--8<--
196196
```
197197

examples/unetformer/train_supervision.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,6 @@ def log_metrics(self, epoch, train_log, val_log):
236236
def main():
237237
args = get_args()
238238
config = py2cfg(args.config_path)
239-
# config = py2cfg("Tran2/config/vaihingen/unetformer.py")
240239
seed_everything(42)
241240
checkpoint_callback = ModelCheckpoint(
242241
save_top_k=config.save_top_k,

0 commit comments

Comments
 (0)