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