Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ImageNet pretrained weights #16

Closed
lonelysheeep opened this issue Apr 17, 2023 · 9 comments
Closed

ImageNet pretrained weights #16

lonelysheeep opened this issue Apr 17, 2023 · 9 comments

Comments

@lonelysheeep
Copy link

I use your source code to reproduce the paper, the result is worse, where <1.25 is only 0.852, is there something wrong with the source code?

@noahzn
Copy link
Owner

noahzn commented Apr 17, 2023

Hi, <1.25=0.852 is abnormal. What PyTorch version do you use? Did you load the pre-trained weights? The source code should work without problems. As far as I know, at least 5 people have reproduced the results.

@lonelysheeep
Copy link
Author

1.7.1,I found that the splits in your code are incomplete, I used the splits in mono2 instead of yours, does this matter?

@noahzn
Copy link
Owner

noahzn commented Apr 17, 2023

Yes. This repo only contains the Eigen test files. You need to prepare the train_files.txt and val_files.txt

@lonelysheeep
Copy link
Author

So may I ask, if I want to reproduce the code to achieve your effect, what else do I need to do besides downloading the code from github and running it according to the readme instructions? For example in the other question it was mentioned that I need to download the pre-trained model or something else?

@noahzn
Copy link
Owner

noahzn commented Apr 17, 2023

Yes, if you want to use the ImageNet pre-trained weights you need to download the pre-trained model and load it by setting --mypretrain

@lonelysheeep
Copy link
Author

Thank you for your answer, I will try again

@lonelysheeep
Copy link
Author

Can you provide me with the download path for the ImageNet pre-training model? I've downloaded several but they don't seem to work

@noahzn
Copy link
Owner

noahzn commented Apr 18, 2023

You can find all the download links you need in the README.md file.

@noahzn noahzn changed the title My training results are not as good as those in your paper ImageNet pretrained weights Apr 18, 2023
@lonelysheeep
Copy link
Author

thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants