Delete resnet50.yaml
Browse files- resnet50.yaml +0 -24
resnet50.yaml
DELETED
@@ -1,24 +0,0 @@
|
|
1 |
-
DATASET:
|
2 |
-
data_path: "./data/head/experiment_5/prop_0_0/"
|
3 |
-
train_path: "./data/head/experiment_5/prop_0_0/train_80"
|
4 |
-
val_path: "./data/head/experiment_5/prop_0_0/val_10"
|
5 |
-
test_path: "./data/head/experiment_5/prop_0_0/test_10"
|
6 |
-
num_class: 4
|
7 |
-
img_size: 299
|
8 |
-
|
9 |
-
MODEL:
|
10 |
-
name: "resnet50"
|
11 |
-
ckpt_path: "./ckpt/head/experiment_5/"
|
12 |
-
model_path: "./models/head/experiment_5/"
|
13 |
-
preds_path: "./predictions/head/experiment_5/"
|
14 |
-
plots_path: "./plots/head/experiment_5/"
|
15 |
-
|
16 |
-
TRAIN:
|
17 |
-
batch_size: 32
|
18 |
-
num_epoch: 500
|
19 |
-
augment: True
|
20 |
-
tune: True
|
21 |
-
optimizer: "Adam"
|
22 |
-
lr: 0.001
|
23 |
-
lr_tune: 0.0001
|
24 |
-
es: 20
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|