Sketchypath

class CNNEncoder(nn.Module): def __init__(self): super(CNNEncoder, self).__init__() self.conv1 = nn.Conv2d(1, 64, kernel_size=3) self.conv2 = nn.Conv2d(64, 128, kernel_size=3) self.fc1 = nn.Linear(128*28*28, 128)

class PathRenderer(nn.Module): def __init__(self): super(PathRenderer, self).__init__() sketchypath

def forward(self, z): z = torch.relu(self.fc1(z)) z = z.view(-1, 128, 28, 28) x = torch.relu(self.conv1(z)) x = torch.sigmoid(self.conv2(x)) return x class CNNEncoder(nn