Train depth / segmentation head

[nikonikolov]

I want to retrain a new depth / segmentation heads. Is the code for training just the head published? I can see the evaluation scripts and the depth / seg inference notebooks, but I am struggling to find the relevant part for training just the heads.