Hi, I used the following code:
model = torchvision.models.resnet50(pretrained=True)
model.cuda()
Then it returned an error message saying that cuda out of memory. Can anybody tell me how to fix this problem?