backbone_evaluation_results.json

[
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "transformer",
        "validation_mse": 0.21577325425086877,
        "validation_mae": 0.36228722945237773,
        "test_mse": 0.21746371760964395,
        "test_mae": 0.36353210285305976,
        "model_path": "test_models/efficientnet_b3_transformer_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "transformer",
        "validation_mse": 0.24033201676912797,
        "validation_mae": 0.42209602166444826,
        "test_mse": 0.19470563121140003,
        "test_mae": 0.37649240642786025,
        "model_path": "test_models/efficientnet_b0_transformer_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "transformer",
        "validation_mse": 0.22672857019381645,
        "validation_mae": 0.3926378931754675,
        "test_mse": 0.22427306957542897,
        "test_mae": 0.39585837423801423,
        "model_path": "test_models/resnet50_transformer_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.2967155438203078,
        "validation_mae": 0.3850937023376807,
        "test_mse": 0.36476454623043536,
        "test_mae": 0.425818096101284,
        "model_path": "test_models/resnet50_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.5120524473679371,
        "validation_mae": 0.5665570046657171,
        "test_mse": 0.5059382550418376,
        "test_mae": 0.555050653219223,
        "model_path": "test_models/efficientnet_b0_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.8020018790012751,
        "validation_mae": 0.7953977386156718,
        "test_mse": 0.7042828559875488,
        "test_mae": 0.7441241115331649,
        "model_path": "test_models/efficientnet_b3_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "gru",
        "validation_mse": 1.1340507984161377,
        "validation_mae": 0.8290961503982544,
        "test_mse": 0.9705999374389649,
        "test_mae": 0.7704607486724854,
        "model_path": "test_models/efficientnet_b0_gru_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "lstm",
        "validation_mse": 2.787272185087204,
        "validation_mae": 1.5404645502567291,
        "test_mse": 2.901867628097534,
        "test_mae": 1.5843785762786866,
        "model_path": "test_models/efficientnet_b0_lstm_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "gru",
        "validation_mse": 3.9335442543029786,
        "validation_mae": 1.8762320041656495,
        "test_mse": 3.72695152759552,
        "test_mae": 1.8381730556488036,
        "model_path": "test_models/resnet50_gru_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "lstm",
        "validation_mse": 6.088638782501221,
        "validation_mae": 2.3887929677963258,
        "test_mse": 6.1847597599029545,
        "test_mae": 2.418113374710083,
        "model_path": "test_models/resnet50_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "gru",
        "validation_mse": 104.58460273742676,
        "validation_mae": 10.183499813079834,
        "test_mse": 104.58482055664062,
        "test_mae": 10.180697345733643,
        "model_path": "test_models/efficientnet_b3_gru_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "lstm",
        "validation_mse": 105.40057525634765,
        "validation_mae": 10.221695899963379,
        "test_mse": 105.17274551391601,
        "test_mae": 10.21053056716919,
        "model_path": "test_models/efficientnet_b3_lstm_model.pt"
    }
]