search

andrearosasco / hyperpcr

4 stars 1 forks source link

Question about how to train this net #1

Closed ran894694447 closed 1 year ago

ran894694447 commented 1 year ago

Hello!Can you tell me how to train this network? For example, which file to run or what environment to configure?

andrearosasco commented 1 year ago

Hi, sorry but I was busy with a demo and didn't really check the notifications here. Did you manage to run it?

ran894694447 commented 1 year ago

Thanks for your reply, I successfully trained the model via train.py.

ran894694447 commented 1 year ago

Hi, I have one more question, when training the backbone of the network, did you use a pre-trained model? Because I noticed that in your paper the learning rate is 1e-4,but in your code the learning rate is 1e-5.I'm looking forward to your replay!

andrearosasco commented 1 year ago

Hi there, the learning rate is indeed 1e-4, here's the config: https://github.com/andrearosasco/confidence_shape_completion/blob/main/configs/main_config.py

ran894694447 commented 1 year ago

Thanks for the reply, I will check it out！