|
a |
|
b/conf/testing.yaml |
|
|
1 |
# Experiment config |
|
|
2 |
task: testing |
|
|
3 |
#experiment_code: EFFDET_002 |
|
|
4 |
results_dir: ../testing_results/012_1/ |
|
|
5 |
weights_path: ./experiments/EFFDET_002/best_weights.pt |
|
|
6 |
# Get dataset and dataloaders |
|
|
7 |
data: |
|
|
8 |
name: IHTestDataset |
|
|
9 |
path: ../patients_windowed/012_1/ |
|
|
10 |
classes: [noIH, IH] |
|
|
11 |
num_workers: 4 |
|
|
12 |
|
|
|
13 |
# Initialize model |
|
|
14 |
model: |
|
|
15 |
name: efficientdet_d0 |
|
|
16 |
feature_extract: False #False, finetune the whole model. True, only update new layer params |
|
|
17 |
use_pretrained: False |
|
|
18 |
print_model: False |