File size: 1,856 Bytes
f0827f9 |
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 |
{
"tensorboard": true,
"wandb": true,
"exp_name": "common0518_3_cnnr101_MNIST_split5_2",
"cfg": "configs/MNIST/cnnr101/common0518_3.yaml",
"n_gpu": 1,
"model_arch": "cnnr101",
"checkpoint_path": "runs/train/common0518_3_cnnr101_MNIST_split5_1_exp/ckpts/best.ckpt",
"image_size": 128,
"batch_size": 256,
"num_workers_pgpu": 10,
"max_epochs": 300,
"warmup_epochs": 10,
"lr": 1e-05,
"lr_strategy": "cosine",
"num_restarts": 2,
"wd": 0.0001,
"data_dir": "./data",
"dataset": "MNIST",
"num_classes": 6,
"known_classes": [
6,
3,
4,
9,
8,
2
],
"osr_dataset": null,
"osr_classes": null,
"random_seed": 2023,
"eval": false,
"opt": "SGD",
"save_freq": 50,
"label_smoothing": 0,
"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",
"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
},
"tensorboard_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/tb/",
"checkpoint_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/ckpts/",
"embedding_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/embeddings/",
"lightning_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp/lightning_logs/",
"result_dir": "runs/train/common0518_3_cnnr101_MNIST_split5_2_exp"
} |