| { |
| "model_type": "sheet_strip", |
| "fine": { |
| "depth": 144, |
| "width": 128, |
| "behind": 72, |
| "spacing": 0.5, |
| "height": null |
| }, |
| "hidden": 128, |
| "n_future": 35, |
| "future_step": 1.0, |
| "gate_plane": 16, |
| "n_history": 128, |
| "max_recovery_distance": 6.0, |
| "patch_radius": 1.0, |
| "recurrent_refinement_steps": 1, |
| "frame_checkpoint": "/home/sean/Documents/villa4/vesuvius/src/vesuvius/neural_tracing/tracers/fibers/output/heading_model_l0_w16_centered_frame/heading_model.pt", |
| "frame_checkpoint_sha256": "c10e21e62d05228bdadeda65334d5c9eede91d942b80ca9d318616eda7cbf35f", |
| "crop_axis": "prediction", |
| "layers": 4, |
| "heads": 4, |
| "ffn": 256, |
| "cnn_channels": [ |
| 12, |
| 24, |
| 64 |
| ], |
| "cnn_blocks": [ |
| 1, |
| 2, |
| 2 |
| ], |
| "cnn_strides": [ |
| 2, |
| 2, |
| 1 |
| ], |
| "cnn_stem": false, |
| "cnn_activation": "leaky_relu", |
| "cell_tokens": "polar", |
| "patch_size": 8, |
| "patch_points": 2, |
| "polar_tube_radius": 3.0, |
| "polar_ring_width": 8.0, |
| "polar_sectors": 6, |
| "polar_ring_growth": 1.5, |
| "polar_ring_widths": [ |
| 3.0, |
| 4.0, |
| 6.0 |
| ], |
| "polar_pool_spacing": 4.0, |
| "multiscale_fine_width": null, |
| "multiscale_coarse_pool": 2, |
| "multiscale_coarse_channels": [ |
| 8, |
| 16, |
| 32 |
| ], |
| "cnn_context_channels": 64, |
| "qk_norm": false, |
| "qk_logit_scale": false, |
| "scoring_pass": true, |
| "tube_channel": true, |
| "tube_sigma": 1.0, |
| "path_lateral_limit": 24.75, |
| "strip_half_width": 30.0, |
| "strip_v_spacing": 2.5, |
| "strip_f_points": 2, |
| "strip_scale": 8.0, |
| "strip_refinement_steps": 1, |
| "strip_full_fine": false, |
| "strip_profile_reach": 6.0, |
| "strip_band_half_height": 0.0, |
| "strip_band_cell": 8, |
| "strip_orientation": true, |
| "strip_uncertainty": true |
| } |
|
|
