-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_config.json
39 lines (39 loc) · 1.16 KB
/
train_config.json
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
{
"general":{
"device":"cuda",
"seed":0,
"scene":"outdoor",
"dataset":["vkitti", "kitti"],
"weather":"cloudy",
"stage":"crossdomain",
"method":"meanteacher",
"neuromorphic":"senet",
"warmup":true,
"warmup_step":4,
"modality":"spike",
"batch_size":8,
"dmax":80,
"alpha":0.999,
"beta":0.01,
"model_type":"hybrid",
"epochs":20,
"loss_depth":"sig",
"loss_uncertainty":"l1",
"loss_distillation":"mse",
"optim":"adam",
"lr_backbone":1e-5,
"lr_scratch":1e-4,
"momentum":0.9,
"save_model":true,
"path_model":"checkpoints/backbone_hybrid_gpu_1_bsz_8_vkitti_spike_kitti_crossdomain",
"path_rgb_pretrain": "checkpoints/backbone_hybrid_gpu_1_bsz_16_vkitti_rgb_imagenet_pretrain/DPTDepthModel_best.pt",
"path_spike_pretrain": "checkpoints/backbone_hybrid_gpu_1_bsz_8_vkitti_spike_kitti_crossmodality/DPTDepthModel_best.pt"
},
"wandb":{
"enable":false,
"username":"username",
"images_to_show":1,
"im_h":540,
"im_w":980
}
}