{ "tensorboard": true, "wandb": true, "exp_name": "common0605_1_vitb16_CUB_split1_2", "cfg": "configs/CUB/vitb16_common0605_1.yaml", "random_seed": 2023, "n_gpu": 1, "model_arch": "vitb16", "image_size": 448, "checkpoint_path": "runs/train/common0605_1_vitb16_CUB_split1_1_exp/ckpts/best.ckpt", "data_dir": "./data", "dataset": "CUB", "num_classes": 100, "known_classes": [ 150, 70, 34, 178, 199, 131, 129, 147, 134, 11, 26, 93, 95, 121, 123, 99, 149, 167, 18, 31, 69, 198, 116, 158, 126, 17, 5, 179, 111, 163, 184, 81, 174, 42, 53, 89, 77, 55, 23, 48, 43, 44, 56, 28, 193, 143, 0, 176, 84, 15, 38, 154, 141, 190, 172, 124, 189, 19, 80, 157, 12, 9, 79, 30, 94, 67, 197, 97, 168, 137, 119, 76, 98, 88, 40, 106, 171, 87, 166, 186, 27, 51, 144, 135, 161, 64, 177, 7, 146, 61, 50, 162, 133, 82, 39, 74, 72, 91, 196, 136 ], "osr_dataset": null, "osr_classes": null, "batch_size": 16, "num_workers_pgpu": 10, "max_epochs": null, "warmup_epochs": null, "train_steps": 6000, "warmup_steps": 500, "lr": 0.008, "lr_strategy": "one_cycle", "num_restarts": null, "wd": 1e-05, "eval": false, "opt": "SGD", "save_freq": 50, "label_smoothing": 0, "transform": "default", "sim_block": "attention", "num_sim_block": 1, "sim_task": "reg", "plot_model": false, "plot_confusion_matrix": false, "plot_feature_embedding": false, "plot_roc": false, "sim_alpha": 0.01, "class_anchor": "mean", "anchor_freq": 10, "fusion": "cross-attention", "model_config": { "sim_block": "attention", "num_sim_block": 1, "sim_task": "reg", "patch_size": 16, "emb_dim": 768, "mlp_dim": 3072, "num_heads": 12, "sim_num_heads": 12, "num_layers": 12, "attn_dropout_rate": 0.0, "dropout_rate": 0.1 }, "tensorboard_dir": "runs/train/common0605_1_vitb16_CUB_split1_2_exp11/tb/", "checkpoint_dir": "runs/train/common0605_1_vitb16_CUB_split1_2_exp11/ckpts/", "embedding_dir": "runs/train/common0605_1_vitb16_CUB_split1_2_exp11/embeddings/", "lightning_dir": "runs/train/common0605_1_vitb16_CUB_split1_2_exp11/lightning_logs/", "result_dir": "runs/train/common0605_1_vitb16_CUB_split1_2_exp11" }