How much memory is required during training and inference?
Closed this issue · 1 comments
zhwis commented
Thanks. The computer stops responding when I take the second step
jiwoon-ahn commented
Not much memory is required on a VGG-16 based network, but you still need a decent GPU device. In my setting, it takes less than 4GB GPU memory.