Hugging Face's logo

Jankiny
/
FCTL-report

Model card Files
FCTL-report
File size: 2,594 Bytes 
6fcb242
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
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"
}