CD-Models / configs /models /tinycd.yaml
Dineth Perera
Publish tested dataset winners and benchmark rankings
ce209f5
Raw
History Blame Contribute Delete
220 Bytes
name: tinycd
num_epochs: 200
optimizer: adamw
lr: 0.0001
weight_decay: 0.01
scheduler: cosine
loss: bce_dice
backbone: EfficientNet-B4
pretrained_weights: model_repos/TinyCD/pretrained/efficientnet_b4.pth
img_size: 256