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