[f45789]: / conf / testing.yaml

Download this file

19 lines (17 with data), 480 Bytes

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