Mapper / mapper /conf /pretrain_resnet.yaml
Cherie Ho
Initial upload
fd01725
raw
history blame
509 Bytes
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