| { |
| "multimodal": { |
| "checkpoint_file": "checkpoints/resnet18_transformer_text_v1_best.pt", |
| "epoch": 10, |
| "num_state_keys": 174, |
| "args": { |
| "train_ids": "datasets/rescue_deepcad_latent/full_v0_len60_excluding_rescue_test/train_ids.txt", |
| "test_ids": "datasets/rescue_deepcad_latent/full_v0_len60_excluding_rescue_test/test_ids.txt", |
| "train_latent_root": "datasets/rescue_deepcad_latent/latents/full_v0_len60_excluding_rescue_test_fp16/train", |
| "test_latent_root": "datasets/rescue_deepcad_latent/latents/overlap_deep_first_len60_trainplusval_fp16/test", |
| "train_text_root": "datasets/rescue_deepcad_latent/text_emb/full_v0_len60_excluding_rescue_test_bert_base_uncased/train", |
| "test_text_root": "datasets/rescue_deepcad_latent/text_emb/full_v0_len60_excluding_rescue_test_bert_base_uncased/test", |
| "init_image_checkpoint": "runs/deepcad_latent/resnet18_transformer_v1_ddp2/best.pt", |
| "freeze_image_epochs": 10, |
| "text_dropout": 0.3, |
| "data_root": "datasets/dataset_v0", |
| "lmdb_path": "cad_data.lmdb", |
| "output_dir": "runs/deepcad_latent/resnet18_transformer_text_v1_ddp2", |
| "device": "cuda", |
| "backbone": "resnet18", |
| "freeze_backbone": false, |
| "fusion_type": "transformer", |
| "transformer_layers": 2, |
| "transformer_heads": 8, |
| "transformer_dropout": 0.1, |
| "img_size": 224, |
| "n_views": 8, |
| "batch_size": 64, |
| "num_workers": 8, |
| "epochs": 10, |
| "lr": 5e-05, |
| "weight_decay": 0.0001, |
| "log_interval": 50, |
| "seed": 42 |
| }, |
| "metrics": { |
| "epoch": 10, |
| "train_mse": 0.06137126215508291, |
| "train_cosine": 0.6658818343481017, |
| "test_mse": 0.05932070141143894, |
| "test_cosine": 0.685916566797592 |
| } |
| }, |
| "image_only": { |
| "checkpoint_file": "checkpoints/resnet18_transformer_image_v1_best.pt", |
| "epoch": 12, |
| "num_state_keys": 156, |
| "args": { |
| "train_ids": "datasets/rescue_deepcad_latent/full_v0_len60_excluding_rescue_test/train_ids.txt", |
| "test_ids": "datasets/rescue_deepcad_latent/full_v0_len60_excluding_rescue_test/test_ids.txt", |
| "train_latent_root": "datasets/rescue_deepcad_latent/latents/full_v0_len60_excluding_rescue_test_fp16/train", |
| "test_latent_root": "datasets/rescue_deepcad_latent/latents/overlap_deep_first_len60_trainplusval_fp16/test", |
| "data_root": "datasets/dataset_v0", |
| "lmdb_path": "cad_data.lmdb", |
| "output_dir": "runs/deepcad_latent/resnet18_transformer_v1_ddp2", |
| "device": "cuda", |
| "backbone": "resnet18", |
| "freeze_backbone": false, |
| "fusion_type": "transformer", |
| "transformer_layers": 2, |
| "transformer_heads": 8, |
| "transformer_dropout": 0.1, |
| "img_size": 224, |
| "n_views": 8, |
| "batch_size": 64, |
| "num_workers": 8, |
| "epochs": 30, |
| "lr": 5e-05, |
| "weight_decay": 0.0001, |
| "resume": "runs/deepcad_latent/resnet18_transformer_v1_ddp2/best.pt", |
| "log_interval": 50, |
| "seed": 42 |
| }, |
| "metrics": { |
| "epoch": 12, |
| "train_mse": 0.06099382114594398, |
| "train_cosine": 0.6691187264323489, |
| "test_mse": 0.06041407727973485, |
| "test_cosine": 0.6800339086372965 |
| } |
| } |
| } |
