upload_metadata.json

{
  "original_checkpoint": "outputs/hf_training/checkpoints/final_model.pth",
  "upload_date": "2025-06-27T09:05:36.254713",
  "analysis": {
    "checkpoint_keys": [
      "epoch",
      "model_state_dict",
      "optimizer_state_dict",
      "loss"
    ],
    "has_model_state_dict": true,
    "has_optimizer_state_dict": true,
    "has_training_metrics": false,
    "has_val_metrics": false,
    "has_args": false,
    "training_info": {},
    "model_info": {
      "num_parameters": 121890467,
      "parameter_keys": [
        "covis_feature_extractor.feature_extractor.encoder.conv1.1.weight",
        "covis_feature_extractor.feature_extractor.encoder.bn1.weight",
        "covis_feature_extractor.feature_extractor.encoder.bn1.bias",
        "covis_feature_extractor.feature_extractor.encoder.bn1.running_mean",
        "covis_feature_extractor.feature_extractor.encoder.bn1.running_var",
        "covis_feature_extractor.feature_extractor.encoder.bn1.num_batches_tracked",
        "covis_feature_extractor.feature_extractor.encoder.layer1.0.conv1.weight",
        "covis_feature_extractor.feature_extractor.encoder.layer1.0.bn1.weight",
        "covis_feature_extractor.feature_extractor.encoder.layer1.0.bn1.bias",
        "covis_feature_extractor.feature_extractor.encoder.layer1.0.bn1.running_mean"
      ]
    }
  },
  "config": {
    "architecture": "CovisPose",
    "framework": "pytorch",
    "model_type": "pose_estimation",
    "task": "panoramic_pose_estimation",
    "input_size": [
      512,
      1024
    ],
    "backbone": "resnet50",
    "num_transformer_layers": 6,
    "transformer_ffn_dim": 2048,
    "num_classes": null,
    "library_name": "pytorch"
  }
}