Image Classification
annanau commited on
Commit
3520d5f
·
verified ·
1 Parent(s): 26fa967

Delete resnet50.yaml

Browse files
Files changed (1) hide show
  1. 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