InPeerReview commited on
Commit
5132c8d
·
verified ·
1 Parent(s): 061b20f

Delete cdxformer.py

Browse files
Files changed (1) hide show
  1. cdxformer.py +0 -77
cdxformer.py DELETED
@@ -1,77 +0,0 @@
1
- net = 'CDXFormer'
2
- ######################## base_config #########################
3
- epoch = 200
4
- gpus = [0]
5
- save_top_k = 1
6
- save_last = True
7
- check_val_every_n_epoch = 1
8
- logging_interval = 'epoch'
9
- resume_ckpt_path = None
10
- monitor_val = 'val_change_f1'
11
- monitor_test = ['test_change_f1']
12
- argmax = True
13
-
14
- test_ckpt_path = "work_dirs/CLCD_BS4_epoch200/CDXFormer/version_0/ckpts/test/test_change_f1/test_change_f1=0.9289-epoch=2.ckpt"
15
-
16
- exp_name = 'WHUCD_epoch200{}/{}'.format(epoch, net)
17
-
18
- ######################## dataset_config ######################
19
- _base_ = [
20
- './_base_/WHUCD_config.py',
21
- ]
22
- num_class = 2
23
- ignore_index = 255
24
-
25
- ######################### model_config #########################
26
- model_config = dict(
27
- backbone = dict(
28
- type = 'Base',
29
- name = 'Seaformer',
30
- ),
31
- decoderhead = dict(
32
- type = 'CDXLSTM',
33
- channels = [64, 128, 192, 256]
34
- )
35
- )
36
- loss_config = dict(
37
- type = 'myLoss',
38
- loss_name = ['FocalLoss', 'dice_loss'],
39
- loss_weight = [0.5, 0.5],
40
- param = dict(
41
- FocalLoss = dict(
42
- gamma=0,
43
- alpha=None
44
- ),
45
- dice_loss = dict(
46
- eps=1e-7
47
- )
48
- )
49
- )
50
-
51
- ######################## optimizer_config ######################
52
- optimizer_config = dict(
53
- optimizer = dict(
54
- type = 'SGD',
55
- lr = 0.05,
56
- momentum=0.9,
57
- weight_decay = 5e-5,
58
- lr_mode = "single"
59
- ),
60
- scheduler = dict(
61
- type = 'step',
62
- step_size = 100,
63
- gamma = 0.1
64
- )
65
- )
66
-
67
- metric_cfg1 = dict(
68
- task = 'multiclass',
69
- average='micro',
70
- num_classes = num_class,
71
- )
72
-
73
- metric_cfg2 = dict(
74
- task = 'multiclass',
75
- average='none',
76
- num_classes = num_class,
77
- )