jiwoon-ahn/psa

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

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.