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

Issue reproducing the uploaded pretrained model #8

Open
unofgithub opened this issue Mar 1, 2023 · 9 comments
Open

Issue reproducing the uploaded pretrained model #8

unofgithub opened this issue Mar 1, 2023 · 9 comments

Comments

@unofgithub
Copy link

unofgithub commented Mar 1, 2023

Hello!

I have been trying to reproduce the pretrained checkpoint using the uploaded code (model.py, networks.py, the utils, train.py) to train on human3.6m, but it is not successful. Could you please let me know if you could upload the exact code repository that was used to get the uploaded pretrained checkpoint? Thank you so much in advance!

@unofgithub
Copy link
Author

unofgithub commented Mar 1, 2023

Also, could you please let me know approximately what the size of the data you trained on was (i.e. approximately how many pairs of images the pretrained model was trained on)? I tried training the model on around 180,000 pairs, but was not able to reproduce the pretrained checkpoint.

@unofgithub
Copy link
Author

Hello,

Specifically, these are the results I am getting:
image

Could you please let me know how I might fix this, since it does not match the pretrained model? Thank you so much!

@lschmidtke
Copy link
Owner

lschmidtke commented Mar 5, 2023

Hi! The code should produce the same results of the provided checkpoint. How do the reconstructed images look like?
The model was trained on roughly 600k images.

@unofgithub
Copy link
Author

The reconstructed images look like:
image

@lschmidtke
Copy link
Owner

lschmidtke commented Mar 9, 2023

Okay, the reconstructions don't look good. What's your batch size and learning rate? Also, how many epochs are you training for?

@unofgithub
Copy link
Author

unofgithub commented Mar 9, 2023 via email

@lschmidtke
Copy link
Owner

That seems reasonable to me. I will run the code myself with those parameters and I'll get back to you.

@unofgithub
Copy link
Author

Hello,

Could you please let me know when there are updates? Thank you so much!

@lschmidtke
Copy link
Owner

Hi,

I encountered the same issue, I suppose it's due to a newer version of one of the packages. I will try and find the cause.

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