Back to SD 2.1. Mari Train: Optimised single_pass — RGB only (no fusion)
Train: Optimised single_pass — RGB only (no fusion) ffa74438
Corrected RGB-only single-pass ablation. Cross-task fusion and all depth/cross-task objectives remain disabled, but the shared 12-channel encoder and visibility head are trainable (freeze_encoder=false, encoder LR=1.5e-5) so the RGB path can learn to ignore the independently noised depth slot. The unused depth decoder stays at LR 0 and the unused depth head is disabled. Checkpoints are selected by validation RGB PSNR (max).
training cancelled
Output
pending ffa74438
Waiting for SLURM job to start...
Datasets
14Dataset
clear_root_rdsDataset real-drive-sim
Type rgb
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/rds/rgb.zip/Dataset
depth_root_rdsDataset real-drive-sim
Type depth
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/rds/depth.zip/Dataset
intrinsics_root_rdsDataset real-drive-sim
Type intrinsics
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/rds/calibration.zip/Dataset
train_hazy_root_rdsDataset real-drive-sim (gloomy_fixed)
Type rgb
Split train
Path
/cluster/work/igp_psr/drothenpiele/data/rds/gloomy_fixed/foggy_rgb.zip/Dataset
val_hazy_root_rdsDataset real-drive-sim (gloomy_fixed)
Type rgb
Split val
Path
/cluster/work/igp_psr/drothenpiele/data/rds/gloomy_fixed/foggy_rgb.zip/Dataset
clear_root_vkitti2Dataset VKITTI2
Type rgb
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/vkitti_2.0.3_rgb.zipDataset
depth_root_vkitti2Dataset VKITTI2
Type depth
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/vkitti_2.0.3_depth.zipDataset
intrinsics_root_vkitti2Dataset VKITTI2
Type intrinsics
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/vkitti_2.0.3_textgt.zip/Dataset
train_hazy_vkittiDataset VKITTI2 (fog_sensor_aug1)
Type rgb
Split train
Path
/cluster/work/igp_psr/drothenpiele/data/vkitti_next/gpu_2/foggy_rgb.zip/Dataset
val_hazy_vkittiDataset VKITTI2 (fog_sensor_aug1)
Type rgb
Split val
Path
/cluster/work/igp_psr/drothenpiele/data/vkitti_next/gpu_2/foggy_rgb.zip/Split
muses_rgbDataset muses
Type rgb
Split fog_day
Path
/cluster/work/igp_psr/drothenpiele/data/muses/frame_camera_trainvaltest.zip/Split
muses_sparse_depthDataset muses
Type sparse_depth
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/muses/lidar_trainvaltest.zip/Split
muses_intrinsicsDataset muses
Type intrinsics
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/muses/frame_camera_trainvaltest.zip/Split
muses_extrinsicsDataset muses
Type camera_extrinsics
Split full
Path
/cluster/work/igp_psr/drothenpiele/data/muses/frame_camera_trainvaltest.zip/Execution Artifacts
2Published Exports
Launch-Owned Exports
These semantic handles are what downstream pipelines should resolve against. Auto-published exports come from config-template metadata captured on this launch.
Published
This launch does not publish any exports yet.
Raw Artifacts
Parameters
278Typed Parameters
Dataset
clear_root_rdsreal-drive-sim / full
(4)Dataset
depth_root_rdsreal-drive-sim / full
(5)Dataset
intrinsics_root_rdsreal-drive-sim / full
(6)Dataset
train_hazy_root_rdsreal-drive-sim / train
(195)Dataset
val_hazy_root_rdsreal-drive-sim / val
(196)Dataset
clear_root_vkitti2VKITTI2 / full
(1)Dataset
depth_root_vkitti2VKITTI2 / full
(2)Dataset
intrinsics_root_vkitti2VKITTI2 / full
(12)Dataset
train_hazy_vkittiVKITTI2 / train
(216)Dataset
val_hazy_vkittiVKITTI2 / val
(238)Split
muses_rgbmuses / fog_day
(77)Split
muses_sparse_depthmuses / full
(76)Split
muses_intrinsicsmuses / full
(82)Split
muses_extrinsicsmuses / full
(81)Simple Parameters
cliptruecropcenterseed42trackerwandbuse_emafalsegpu_typertx_4090:1job_namesd_train_concatnorm_max1norm_min-1run_namesingle-pass-rgb-only-no-fusiontmp_size10Gadam_8bittruedata_kindreal_drive_simema_decay0.9999ema_dtypefloat32log_every10max_depth800min_depth1adam_beta10.9adam_beta20.999batch_size16clear_root4depth_modemetric_logdepth_root5image_size[384, 768]joint_modesingle_passlog_imagestruenum_epochs200output_dir/cluster/scratch/drothenpiele/SD21/exp_1time_limit2-00:00:00mem_per_cpu8Gnum_workers4adam_epsilon1e-8adam_foreachfalseaspect_ratio[1, 2]camera_modelpinholeconditioningconcatlambda_depth0lambda_point0.0min_lr_ratio0.01project_namejoint-dehazinguse_geometryfalsewarmup_ratio0.05weight_decay0cpus_per_task8lambda_camera0.0lambda_dehaze1lambda_eg_ssi0.0learning_rate0.00004max_depth_rds800max_grad_norm1min_depth_rds0.00001val_hazy_root104camera_enabledtruefreeze_encoderfalselog_visibilitytruenum_log_images4use_depth_headfalseval_batch_size10camera_fallbackinvaliddepth_head_typebase_residualenable_xformerstrueeuler_train_dir/cluster/work/igp_psr/drothenpiele/data/out/train/sd21-jointlambda_depth_tv0mixed_precisionfp16prediction_typev_predictionsource_kind_rdsreal_drive_simtrain_hazy_root103visibility_cmapviridiscamera_embeddingsinecamera_emit_raysfalsecamera_mlp_ratio4camera_num_heads4camera_token_dim256depth_noise_typeannealed_multireslambda_log_depth0.0lr_schedule_typeconstant_with_warmupmin_max_quantile0.02no_decay_enabledfalsepretrained_modelsd2-community/stable-diffusion-2-1sky_depth_meters800camera_num_layers2camera_num_tokens4conv_in_init_modejoint_marigoldgeometry_lambda_z0.15geometry_timestep0joint_weight_psnr1joint_weight_ssim10lambda_depth_head0lambda_lambda_mse0.0lambda_visibility0.05max_depth_vkitti2300min_depth_vkitti20.00001sampling_strategysource_balancedsource_weight_rds1depth_ensemble_tol0.001lambda_uncertainty0.0model_depth_outputvae_decodeval_every_n_epochs1camera_feature_poolavgdepth_ensemble_size1geometry_lambda_phi1num_inference_steps25rds_val_max_samples128sampling_epoch_sizeallsave_every_n_epochs2source_kind_vkitti2vkitti2use_visibility_headtruewarmup_start_factor0.001camera_metadata_rootnullconv_in_target_scale0.7071external_camera_modenonegeometry_loss_robustcharbonnierlambda_vae_depth_rec0lambda_visibility_tv0.005sky_depth_meters_rds800camera_branch_enabledtruecamera_embedding_sizelatentdepth_gradient_robustcharbonnierdepth_multires_levels4encoder_learning_rate0.000015geometry_head_enabledfalsegeometry_lambda_theta1lambda_depth_gradient0lambda_geometry_total0.0source_weight_vkitti21use_cross_task_fusionfalsevae_decoder_trainablefalsevisibility_rank_pairs128weight_decay_backbone0weight_decay_no_decay0zero_grad_set_to_nonetruecamera_feature_sourcesautocamera_metadata_formateuler_loadingdepth_ensemble_max_res1024depth_smoothness_spacelatent_x0freeze_cross_attentiontruegradient_checkpointingtrueinference_depth_outputvae_decodelambda_geo_consistency0.0lambda_visibility_rank0use_task_skip_adaptersfalsevisibility_rank_margin0.05camera_embedding_stride8camera_prediction_spaceresidual_intrinsicsdepth_ensemble_max_iter2depth_multires_strength0.9depth_smoothness_scales4geometric_pairs_enabledfalsejoint_noise_correlation0joint_weight_delta1_pct0.5keep_last_n_checkpoints5validation_depth_outputvae_decodevisibility_target_gamma4visibility_warmup_steps1000vkitti2_val_max_samples0cross_task_fusion_blocks[0,1,2,3]cross_task_fusion_detachfalsecross_task_fusion_kerneladaptivedepth_ensemble_reductionmediandepth_normalization_typelog_depthgeometry_camera_encodingsinegeometry_head_num_blocks2joint_weight_abs_rel_pct0.5sky_depth_meters_vkitti2800camera_intrinsics_loadinghierarchicaldepth_diagnostics_enabledtruedepth_head_residual_scale0.05geometry_depth_conventionz_depthgeometry_metric_depth_max800geometry_metric_depth_min1num_inference_steps_depth20vae_decoder_learning_rate0.000001visibility_apply_to_depthfalsecamera_residual_logit_clip4depth_head_hidden_channels64depth_resize_interpolationnearestdepth_smoothness_normalizetruedetach_rays_for_point_losstrueenable_efficient_attentiontruelambda_visibility_preserve0.05visibility_apply_to_dehazetruevisibility_apply_to_fusionfalsevisibility_hidden_channels32visibility_target_quantile0.9weight_decay_depth_decoder0weight_decay_geometry_head0camera_require_for_trainingtruecamera_sine_num_frequenciesnullcheckpoint_selection_metricpsnrcross_task_fusion_dilations[1,2,4]depth_decoder_learning_rate0emit_camera_full_resolutiontruegeometry_eg_ssi_edge_sourcehazygeometry_encoder_input_modehazy_hazygeometry_head_learning_rate0.00004geometry_output_uncertaintytruegeometry_ray_representationunitgradient_accumulation_steps1validation_geometry_enabledfalseweight_decay_dehaze_decoder0camera_output_representationanglescamera_valid_for_approximatefalsecross_task_fusion_directions["rgb_to_depth","depth_to_rgb"]dehaze_decoder_learning_rate0.00002depth_head_context_dilations[1, 2, 4]depth_smoothness_edge_sourcecleardepth_smoothness_edge_weight10geometry_head_residual_scale0.05lambda_depth_edge_smoothness0lambda_rgb_depth_consistency0weight_decay_geometry_camera0depth_diagnostics_edge_sourcedehazeddepth_diagnostics_edge_weight10depth_head_base_upsample_modebilineargeometry_camera_learning_rate0.00004geometry_head_hidden_channels128geometry_sine_num_frequencies64visibility_decoder_grad_scale1.0camera_approximate_fov_degrees60checkpoint_selection_directionmaxcross_task_fusion_learned_gatetruegeometry_use_deterministic_vaetruenative_crop_lower_region_ratio0depth_multires_downscale_factor2lambda_depth_gt_edge_smoothness0validation_geometry_camera_modepredictedvalidation_geometry_output_modegeometry_headcamera_include_input_in_encodingnullcamera_intrinsics_metadata_scopeintrinsicscross_task_fusion_gate_init_bias0cross_task_fusion_sender_lowpassfalsedepth_head_use_depthwise_contexttruegeometric_pairs_views_per_sample2geometry_condition_on_visibilityfalsegeometry_detach_encoder_featurestruecross_task_fusion_hidden_channels64depth_diagnostics_normalize_depthtruenative_crop_centered_region_ratio0native_crop_lower_region_positioncenterrgb_depth_consistency_edge_weight4rgb_depth_consistency_weight_modehallucinatedgeometric_pairs_emit_pair_metadatatruegeometry_include_input_in_encodingfalselambda_depth_multiscale_smoothness0depth_ensemble_regularizer_strength0.02native_crop_centered_region_positioncentercross_task_fusion_attention_kv_tokens1024cross_task_fusion_attention_max_block-1depth_diagnostics_road_edge_threshold0.08geometry_head_detach_camera_embeddingtruedepth_diagnostics_road_bottom_fraction0.4geometry_encoder_no_grad_when_detachedtruecross_task_fusion_sender_lowpass_kernel3depth_diagnostics_high_frequency_kernel5validation_geometry_compute_edge_metricstruevalidation_geometry_compute_point_metricsfalsevalidation_geometry_max_points_per_sample20000validation_geometry_save_geometry_samplestruevalidation_geometry_compute_camera_metricstruevalidation_geometry_compute_resolution_sweepfalsevalidation_geometry_max_point_metric_samples8validation_geometry_save_point_cloud_samplesfalsevalidation_geometry_compute_uncertainty_metricstrueRaw JSON
{
"clip": "true",
"crop": "center",
"seed": 42,
"tracker": "wandb",
"use_ema": "false",
"gpu_type": "rtx_4090:1",
"job_name": "sd_train_concat",
"norm_max": 1,
"norm_min": -1,
"run_name": "single-pass-rgb-only-no-fusion"...Events
Launch Events
0No launch events recorded.