Upload 2 files

common_cnnr101_FGVC_split1_2_exp/ckpts/best.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7ced36f4f88b6202b33cf6ce03e74481ef7e40f7fa2ae526ec3601a7c44e2ab
+size 556873244

common_cnnr101_FGVC_split1_2_exp/config.json

@@ -0,0 +1,114 @@
+{
+    "tensorboard": true,
+    "wandb": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp/wandb",
+    "exp_name": "common0813_1_cnnr101_FGVC_split1_2",
+    "cfg": "configs/FGVC/cnnr101_common0813_1.yaml",
+    "random_seed": 2023,
+    "n_gpu": 1,
+    "model_arch": "cnnr101",
+    "image_size": 448,
+    "checkpoint_path": "runs/train/common0813_1_cnnr101_FGVC_split1_1_exp/ckpts/best.ckpt",
+    "data_dir": "./data",
+    "dataset": "FGVC",
+    "num_classes": 50,
+    "known_classes": [
+        0,
+        1,
+        2,
+        3,
+        4,
+        5,
+        10,
+        11,
+        14,
+        16,
+        17,
+        19,
+        21,
+        22,
+        23,
+        24,
+        27,
+        28,
+        29,
+        30,
+        33,
+        36,
+        37,
+        38,
+        39,
+        41,
+        43,
+        44,
+        45,
+        46,
+        47,
+        48,
+        52,
+        53,
+        56,
+        57,
+        58,
+        63,
+        64,
+        65,
+        66,
+        67,
+        71,
+        73,
+        76,
+        77,
+        79,
+        92,
+        95,
+        99
+    ],
+    "osr_dataset": null,
+    "osr_classes": null,
+    "batch_size": 16,
+    "num_workers_pgpu": 10,
+    "max_epochs": null,
+    "warmup_epochs": null,
+    "train_steps": 15900,
+    "warmup_steps": 500,
+    "lr": 0.01,
+    "lr_strategy": "one_cycle",
+    "num_restarts": null,
+    "wd": 1e-05,
+    "opt": "SGD",
+    "save_freq": 50,
+    "label_smoothing": 0.1,
+    "transform": "rand-augment",
+    "resume": false,
+    "eval": false,
+    "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,
+    "sim_lambda": 0.01,
+    "frozen_encoder": false,
+    "class_anchor": "mean",
+    "anchor_freq": 30,
+    "fusion": "cross-attention",
+    "model_config": {
+        "sim_block": "attention",
+        "num_sim_block": 1,
+        "sim_task": "reg",
+        "emb_dim": 768,
+        "mlp_dim": 3072,
+        "sim_num_heads": 12,
+        "num_heads": 12,
+        "num_layers": 12,
+        "attn_dropout_rate": 0.0,
+        "dropout_rate": 0.1
+    },
+    "checkpoint_dir": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp/ckpts/",
+    "embedding_dir": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp/embeddings/",
+    "lightning_dir": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp/lightning_logs/",
+    "result_dir": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp",
+    "tensorboard_dir": "runs/train/common0813_1_cnnr101_FGVC_split1_2_exp/tb"
+}