DriverNet/configs/config.yaml at main · DriverNet-Project/DriverNet · 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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
global:
    seed: 42
    out_dir: "output"

trainer:
    max_epochs: 30
    accelerator: auto
    devices: auto
    precision: "bf16-mixed"
    strategy: "auto"

model:
    name: "alexnet_depthg"
    num_classes: 10
    pretrained: true
    lr: 2e-4
    image_size: 384
    weight_decay: 5e-3
    scheduler: "onecycle"
    # warmup_pct: 0.15
    label_smoothing: 0.1
    consistency_loss_type: "mse"
    # temperature: 4.0
    max_logit_norm: 10.0
    ema_decay: 0.999
    ema_warmup_steps: 400
    consistency_rampup_steps: 0.25
    final_consistency_weight: 1.0

data:
    original_data_dir: "../input/original/train"
    depth_data_dir: "../input/processed_depth/train"
    csv_path: "../input/driver_imgs_list.csv"
    batch_size: 128
    num_workers: 12
    pin_memory: True
    persistent_workers: True
    prefetch_factor: 8
    image_size: 384
    validation_split: 0.2
    # num_folds: 3
    # fold_index: 0
    # start_fold: 0

data_test:
    predict_dir: "../input/original/test"
    depth_data_dir: "../input/processed_depth/test"
    batch_size: 512
    num_workers: 8
    pin_memory: True
    persistent_workers: True
    prefetch_factor: 4
    image_size: 384