CycleNet icon indicating copy to clipboard operation
CycleNet copied to clipboard

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

Open yryun opened this issue 4 months ago • 0 comments

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 avatar Mar 01 '24 09:03 yryun