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

Too much VRAM cost and ckpt file volume compared to the paper #6

Open
yryun opened this issue Mar 1, 2024 · 0 comments
Open

Too much VRAM cost and ckpt file volume compared to the paper #6

yryun opened this issue Mar 1, 2024 · 0 comments

Comments

@yryun
Copy link

yryun commented Mar 1, 2024

Thanks for great work!

I followed the example provided with cfill dataset.
-> https://github.com/sled-group/CycleNet/blob/main/docs/train.md

I found 2 differences with the original works.

  1. Training with batch size = 1 cost over 30GB VRAM, which is different in paper (batch size = 4 cost 27.9GB VRAM).

  2. My ckpt file (saved in training) is 13.7 GB, which is different in provided ckpt (summer2winter.ckpt = 6.7 GB)

Would you explain how to reproduce the original works?

@yryun yryun changed the title Too much VRAM and ckpt file compared to example provided Too much VRAM cost and ckpt file volume compared to the paper Mar 1, 2024
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

1 participant