Hi,
Thank you so much for the reply! I tried the method you provided in that other post, it worked in terms of adding dropout layers after all conv2d layers. But when I tried to send the model to GPU I got ‘cuda out of memory’. I’m guessing, is it because of the deepcopy?
And I’m not sure that I understand the first solution you mentioned, could you please provide me with some code examples?
Thank you!