| { |
| "model_type": "camera_orbit_compact_09_2b", |
| "hub_id": "tbhugging/camera_orbit_compact_09_2b", |
| "protocol": "09_2B", |
| "architecture": "pooled_gap", |
| "library_class": "GeometryAwareFourierFusionLocalizer", |
| "factory_method": "for_09_2_pooled", |
| "variant_id": "m09_2_e2e_pooled_geometry_fusion", |
| "fusion_pattern": "e2e_pooled_geometry_fusion", |
| "backbone_kind": "pooled", |
| "n_views": 6, |
| "view_angles_deg": [ |
| 0.0, |
| 60.0, |
| 120.0, |
| 180.0, |
| 240.0, |
| 300.0 |
| ], |
| "geometry_features": [ |
| "sin_theta", |
| "cos_theta" |
| ], |
| "geometry_mode": "concat", |
| "fusion_hidden": 128, |
| "fusion_depth": 1, |
| "n_outputs": 3, |
| "y_fields": [ |
| "particle_x", |
| "particle_y", |
| "particle_z" |
| ], |
| "x_field": "anomaly_ref", |
| "image_normalize": "per_image_zscore", |
| "input_channels": 1, |
| "image_height": 128, |
| "image_width": 128, |
| "lr": 0.001, |
| "n_params": 132358, |
| "source_checkpoint": "checkpoints/m9/m09_e2e_pooled_geometry_fusion.pt", |
| "source_experiment_id": "m09_e2e_pooled_geometry_fusion", |
| "source_state_key": "f2_state", |
| "weights_file": "pytorch_model.bin", |
| "metrics": { |
| "validation_RMSE_total": 1.3202524185180664, |
| "test_RMSE_total": 0.8607456088066101, |
| "n_params": 132358.0, |
| "lr": 0.001 |
| } |
| } |
|
|
