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