Hugging Face's logo

VBoussot
/
ImpactSynth

Model card Files
xet
Community
VBoussot commited on
Commit
7c31387
·
verified ·
1 Parent(s): 85ea07f

Upload Config.yml

Browse files
Files changed (1) hide show
  1. CBCT/Config.yml +158 -0
CBCT/Config.yml ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Trainer:
2
+ Model:
3
+ classpath: Model:UNetpp
4
+ UNetpp:
5
+ schedulers:
6
+ StepLR:
7
+ step_size: 10
8
+ gamma: 0.75
9
+ last_epoch: -1
10
+ verbose: deprecated
11
+ nb_step: 0
12
+ outputs_criterions:
13
+ Head:Tanh:
14
+ targets_criterions:
15
+ CT:
16
+ criterions_loader:
17
+ MAE:
18
+ schedulers:
19
+ Constant:
20
+ nb_step: 0
21
+ value: 1
22
+ is_loss: true
23
+ group: 0
24
+ start: 0
25
+ stop: None
26
+ accumulation: false
27
+ reduction: mean
28
+ IMPACTSynth:
29
+ is_loss: true
30
+ group: 0
31
+ start: 0
32
+ stop: None
33
+ accumulation: false
34
+ model_name: SAM2.1/Small_3_Layers.pt
35
+ shape:
36
+ - 0
37
+ - 0
38
+ in_channels: 3
39
+ weights:
40
+ - 1
41
+ - 1
42
+ - 1
43
+ schedulers:
44
+ Constant:
45
+ nb_step: 0
46
+ value: 1
47
+ CT;MASK:
48
+ criterions_loader:
49
+ MAE:
50
+ schedulers:
51
+ Constant:
52
+ nb_step: 0
53
+ value: 1
54
+ is_loss: false
55
+ group: 0
56
+ start: 0
57
+ stop: None
58
+ accumulation: false
59
+ reduction: mean
60
+ pretrained: false
61
+ Optimizer:
62
+ name: AdamW
63
+ lr: 0.001
64
+ betas:
65
+ - 0.9
66
+ - 0.999
67
+ eps: 1e-08
68
+ weight_decay: 0.001
69
+ amsgrad: false
70
+ maximize: false
71
+ foreach: None
72
+ capturable: false
73
+ differentiable: false
74
+ fused: None
75
+ Dataset:
76
+ groups_src:
77
+ CT:
78
+ groups_dest:
79
+ CT:
80
+ transforms:
81
+ Clip:
82
+ min_value: -1024
83
+ max_value: 3071
84
+ save_clip_min: true
85
+ save_clip_max: true
86
+ mask: None
87
+ Normalize:
88
+ lazy: false
89
+ channels: None
90
+ min_value: -1
91
+ max_value: 1
92
+ inverse: true
93
+ patch_transforms: None
94
+ is_input: false
95
+ CBCT:
96
+ groups_dest:
97
+ CBCT:
98
+ transforms:
99
+ Clip:
100
+ min_value: min
101
+ max_value: percentile:99.5
102
+ save_clip_min: false
103
+ save_clip_max: false
104
+ mask: None
105
+ Normalize:
106
+ lazy: false
107
+ channels: None
108
+ min_value: -1
109
+ max_value: 1
110
+ inverse: true
111
+ patch_transforms: None
112
+ is_input: true
113
+ augmentations:
114
+ DataAugmentation_0:
115
+ data_augmentations:
116
+ Flip:
117
+ f_prob:
118
+ - 0
119
+ - 0.5
120
+ - 0.5
121
+ prob: 1
122
+ nb: 1
123
+ Patch:
124
+ patch_size:
125
+ - 1
126
+ - 320
127
+ - 320
128
+ overlap: None
129
+ mask: None
130
+ pad_value: -1
131
+ extend_slice: 2
132
+ subset: None
133
+ shuffle: true
134
+ filter: None
135
+ dataset_filenames:
136
+ - ./Dataset/:a:mha
137
+ inline_augmentations: true
138
+ use_cache: true
139
+ batch_size: 32
140
+ validation: None
141
+ train_name: FT_0
142
+ manual_seed: 32
143
+ epochs: 100
144
+ it_validation: 2500
145
+ autocast: false
146
+ gradient_checkpoints: None
147
+ gpu_checkpoints: None
148
+ ema_decay: 0
149
+ data_log:
150
+ - CT/IMAGES/5
151
+ - CBCT/IMAGES/5
152
+ - Head:Tanh/IMAGES/5
153
+ save_checkpoint_mode: ALL
154
+ EarlyStopping:
155
+ monitor: []
156
+ patience: 30
157
+ min_delta: 0.0
158
+ mode: min