snnetv2-image-classification / gradio_demo.json
HubHop
update
bcfa144
raw
history blame
780 Bytes
{
"anchors": ["deit_small_patch16_LS", "deit_large_patch16_LS"],
"batch_size": 64,
"snnet_name": "snnet_v2",
"data_path": "/data2/datasets/imagenet",
"data_set": "IMNET",
"exp_name": "stitch_s_l_v2_lora_r_64_50_ep",
"input_size": 224,
"num_workers": 10,
"lr": 0.00003,
"warmup_lr": 1e-7,
"epochs": 50,
"weight_decay": 0.02,
"sched": "cosine",
"eval_crop_ratio": 1.0,
"reprob": 0.0,
"smoothing": 0.1,
"warmup_epochs": 5,
"drop": 0.0,
"seed": 0,
"opt": "fusedlamb",
"mixup": 0,
"anchor_drop_path": [0.05, 0.4],
"cutmix": 1.0,
"color_jitter": 0.3,
"unscale_lr": true,
"no_repeated_aug": true,
"ThreeAugment": true,
"src": true,
"lora_r": 64,
"pretrained_deit": "../pretrained_weights",
"resume": "snnetv2_deit3_s_l.pth"
}