HPST-Nova/options.json at main · dikshantsagar/HPST-Nova · 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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
{
    "pointnet_patch_embed_depth": 1,
    "pointnet_patch_embed_channels": 512,
    "pointnet_num_heads": 16,
    "pointnet_patch_embed_groups": 6,
    "pointnet_patch_embed_neighbours": 8,
    "pointnet_enc_depths": [
        2,
        6,
        2
    ],
    "pointnet_enc_channels": [
        192,
        384,
        512
    ],
    "pointnet_enc_groups": [
        24,
        48,
        64
    ],
    "pointnet_enc_neighbours": [
        16,
        16,
        16
    ],
    "pointnet_dec_depths": [
        1,
        1,
        1
    ],
    "pointnet_dec_channels": [
        96,
        192,
        384
    ],
    "pointnet_dec_groups": [
        12,
        24,
        48
    ],
    "pointnet_dec_neighbours": [
        16,
        16,
        16
    ],
    "pointnet_grid_sizes": [
        16,
        32,
        64
    ],
    "dropout": 0.0,
    "training_file": "/baldig/physicsprojects2/dikshans/datasets/nova/preprocessed_nova_miniprod6_1_cvnlabmaps.h5",
    "testing_file": "",
    "validation_file": "",
    "dataset_limit": 1.0,
    "train_validation_split": 0.85,
    "batch_size": 220,
    "num_dataloader_workers": 16,
    "normalize_features": 1,
    "optimizer": "AdamW",
    "learning_rate": 0.013046563604508194,
    "l2_penalty": 0.05,
    "gradient_clip": 42.0,
    "epochs": 72,
    "learning_rate_warmup_epochs": 4.0,
    "learning_rate_cycles": 4,
    "num_gpu": 8,
    "event_prong_loss_proportion": 0.5,
    "loss_beta": 0.9999999,
    "loss_gamma": 2.0,
    "pixel_noise_std": 0.01,
    "verbose_output": false,
    "usable_gpus": "",
    "trial_time": "",
    "trial_output_dir": "./test_output"
}