Spaces:
Sleeping
Sleeping
File size: 454 Bytes
e2855de 2add545 e2855de 9a6ea32 e2855de 9a6ea32 e2855de 5606920 9a6ea32 5606920 e2855de |
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 |
model:
backbone: resnet50
num_classes: 2
pretrained: true
learning_rate: 0.0050
epochs: 10
batch_size: 8
optimizer: adam
image_sample_size: 10000
trainable_backbone_layers: 1
dataset:
image_dir: datasets/images
annotation_dir: datasets/annotations
train_split: 0.8
val_split: 0.2
notebooks:
visualization: logs/training_log.json
logs:
log_dir: logs/training_log.txt
log_interval: 10
evaluation:
iou_threshold: 0.5
|