Skip to content

Commit 71c5c19

Browse files
committed
update hrnet v2
1 parent 215617d commit 71c5c19

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

config/ade20k-hrnetv2.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ MODEL:
1616

1717
TRAIN:
1818
batch_size_per_gpu: 2
19-
num_epoch: 20
19+
num_epoch: 30
2020
start_epoch: 0
2121
epoch_iters: 5000
2222
optim: "SGD"
@@ -33,10 +33,10 @@ TRAIN:
3333

3434
VAL:
3535
visualize: False
36-
checkpoint: "epoch_20.pth"
36+
checkpoint: "epoch_30.pth"
3737

3838
TEST:
39-
checkpoint: "epoch_20.pth"
39+
checkpoint: "epoch_30.pth"
4040
result: "./"
4141

4242
DIR: "ckpt/ade20k-hrnetv2-c1"

0 commit comments

Comments
 (0)