search

prs-eth / Marigold

[CVPR 2024 - Oral, Best Paper Award Candidate] Marigold: Repurposing Diffusion-Based Image Generators for Monocular Depth Estimation
https://marigoldmonodepth.github.io
Apache License 2.0
2.42k stars 132 forks source link

Where is the core code for 'duplicate the weight tensor of the input layer'? #118

Closed Kyfafyd closed 2 months ago

Kyfafyd commented 2 months ago

Thanks for your great work! I would like to learn where is the core code for duplication?

markkua commented 2 months ago

Hi, it's here: https://github.com/prs-eth/Marigold/blob/62413d56099d36573b2de1eb8c429839734b7782/src/trainer/marigold_trainer.py#L169