fanurs

# When to reinstall?
# - When VS Code cannot connect to remote server via the SSH extension
# - New extensions cannot be installed on remote server

# 1. Connect to remote server (without using VS Code, of course).

# 2. Remove the ~/.vscode-server directory
cd ~
rm -rf .vscode-server/

alper111

import torch
import torchvision

class VGGPerceptualLoss(torch.nn.Module):
    def __init__(self, resize=True):
        super(VGGPerceptualLoss, self).__init__()
        blocks = []
        blocks.append(torchvision.models.vgg16(pretrained=True).features[:4].eval())
        blocks.append(torchvision.models.vgg16(pretrained=True).features[4:9].eval())
        blocks.append(torchvision.models.vgg16(pretrained=True).features[9:16].eval())