Because ur batch-size is 200 and densenet is demanding.  Why don’t you use smaller BS and increase up the the gpu memory?
I mean, according to the code your batch size is 200.
Densenet is heavy (densily conected)