File size: 509 Bytes
fd01725 |
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 |
defaults:
- schema/data: mia
- data: mia
- model: mapper
- training
- _self_
- override model/schema/backbone: resnet
- override model/image_encoder: resnet
experiment:
name: MIA_DINOv2_Pretrain
model:
loss:
xent_weight: 1.0
dice_weight: 1.0
focal_loss: false
focal_loss_gamma: 2.0
requires_frustrum: true
class_weights: [ 1.00351229, 4.34782609, 1.00110121, 1.03124678,
6.69792364, 7.55857899 ]
training:
trainer:
max_steps: 10
max_epochs: 15 |