shariqfarooq123 / AdaBins

Official implementation of Adabins: Depth Estimation using adaptive bins
GNU General Public License v3.0
730 stars 156 forks source link

A question on weight decay. #42

Closed kmbmjn closed 2 years ago

kmbmjn commented 3 years ago

Hi, Thank you for sharing the code. I would like to ask a question on the weight decay coefficient.

In your paper, you said, "For training, we use the AdamW optimizer [30] with weight-decay 10−2." However, in your code(args_train_kitti_eigen.txt or args_train_nyu.txt), you wrote, "--wd 0.1" Can you tell me which is correct?

Thanks.

shariqfarooq123 commented 2 years ago

The final model provided in this repo is with wd of 0.1