-
Notifications
You must be signed in to change notification settings - Fork 2
/
vits16-coco_args.json
1 lines (1 loc) · 1.18 KB
/
vits16-coco_args.json
1
{"image_queries": ".", "labels_path": ".", "arch": "vit_small", "resnet_dilate": 2, "patch_size": 16, "out_dim": 65536, "out_dim_c": 65536, "norm_last_layer": false, "momentum_teacher": 0.996, "use_bn_in_head": false, "warmup_teacher_temp": 0.04, "teacher_temp": 0.07, "warmup_teacher_temp_epochs": 30, "warmup_teacher_temp_c": 0.04, "teacher_temp_c": 0.07, "student_temp_c": 0.1, "use_fp16": true, "weight_decay": 0.04, "weight_decay_end": 0.4, "clip_grad": 0.0, "batch_size_per_gpu": 32, "epochs": 300, "freeze_last_layer": 1, "lr": 0.0005, "warmup_epochs": 10, "min_lr": 1e-05, "optimizer": "adamw", "drop_path_rate": 0.1, "global_crops_scale": [0.25, 1.0], "local_crops_number": 0, "local_crops_scale": [0.05, 0.25], "output_dir": ".", "saveckp_freq": 100, "seed": 0, "num_workers": 7, "dist_url": ".", "local_rank": 0, "sinkhorn_lambda": 20.0, "sinkhorn_iterations": 5, "student_temp": 1.0, "dataset_type": "coco", "imagenet1k_path": ".", "coco_path": ".", "data_path": ".", "untar_path": ".", "pos_alpha": [2.0, 2.0], "which_features": "last", "n_tokens": 64, "same_teacher_augmentations": false, "start_checkpoint_path": "", "queue_size": 25000, "n_iter": 5, "update_c": true, "normalized_matching": true}