ImageClassification/config.toml at master · matei8/ImageClassification · GitHub

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
26
27
28
29
30
31
32
33
34
35
36
37
[experiment]
name = "landpatches_mlp_baseline"
base_results_dir = "results"
use_timestamp = false

[dataset]
selected = "imagebits"
base_path = "data"
img_size = [64, 64]  # Rezoluție comună (Land patches e 64, Imagebits e 96 -> facem resize la 64)

[preprocessing]
mean = [0.3483, 0.3846, 0.4113]
std  = [0.0921, 0.0663, 0.0561]

[steps]
run_eda = false
run_training = true
run_evaluation = true
save_results = true

[augmentation]
active = true
description = "HorizontalFlip(p=0.5), RandomBrightnessContrast(p=0.2)"

[model]
type = "MLP"
epochs = 35
batch_size = 64
learning_rate = 0.0005
device = "cuda"
pretrained_path = ""

dropout_rate = 0.5
weight_decay = 0.0005
label_smoothing = 0.1
early_stopping_patience = 7
scheduler_patience = 5