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

The model size is extremely large T_T #4

Closed
1165048017 opened this issue May 15, 2019 · 1 comment
Closed

The model size is extremely large T_T #4

1165048017 opened this issue May 15, 2019 · 1 comment

Comments

@1165048017
Copy link

I trained the original model without any modification, and got a model which can work well. But the model size is too big, about 1.1GB, while the supplied pretrained model is 217MB.
I just wonder if the pretrained model has discard some weights? Like the discriminator network's weights?

@tancik
Copy link
Owner

tancik commented May 15, 2019

Good catch. The large size was due to saving the optimizer variables. The code has been updated to only save constant values. The model size is now ~230mb,

@tancik tancik closed this as completed May 15, 2019
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