Hello folks,
Actually I have a requirement that convert keras model to pytorch, and I did it. But when am training keras model it is consuming 100MB of GPU memory and with the same data set when am training in pytorch it almost consuming 1GB of GPU memory, even i didn’t change the data set.
When i set pytorch model to(device), occupying 800MB of GPU memory but this is not happening with keras
appreciate for the quick response.
Thanks,
Vamshi