ujscjj / DPTNet

100 stars 23 forks source link

Question about the enc_dim parameter. #4

Open yaozengwei opened 3 years ago

yaozengwei commented 3 years ago

Thanks for sharing your codes. I'm wondering if this enc_dim is 64 (as in your paper) or 256 (as in the code page https://github.com/ujscjj/DPTNet/blob/master/dpt_net.py).

ujscjj commented 2 years ago

The enc_dim is 256 in encoder, and the translate it to 64 by a linear layer before the dpt net.