Did you use BatchNorm in your DPTHead? - Githubissues

naver / dust3r

DUSt3R: Geometric 3D Vision Made Easy

https://dust3r.europe.naverlabs.com/

Other

4.65k stars 515 forks source link

Did you use BatchNorm in your DPTHead? #61

Closed LiZizun closed 2 months ago

yocabon commented 2 months ago

Hi,

the dpt head is instantiated here: https://github.com/naver/dust3r/blob/main/dust3r/heads/dpt_head.py#L96

in particular, we use head_type='regression' and use_bn is not set to True so BatchNorm is not used in our DPTHead

--> the rest of the instantiation code is here https://github.com/naver/croco/blob/743ee71a2a9bf57cea6832a9064a70a0597fcfcb/models/dpt_block.py#L289