[f45789]: / conf / evaluation.yaml

Download this file

21 lines (19 with data), 450 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
# Experiment config
task: evaluation
experiments_dir: ./experiments
experiment_code: EFFDET_003
metrics: False
grad_cam: True
# Get dataset and dataloaders
data:
name: IHDataset
path: ../data/windowed/
classes: [noIH, IH]
num_workers: 8
batch_size: 8
# Initialize model
model:
name: efficientdet_d0
feature_extract: False #False, finetune the whole model. True, only update new layer params
use_pretrained: True
print_model: False