def forward(self, x): z = self.encoder(x) graphics = self.decoder(z) svg = self.path_renderer(graphics) return svg
class SketchyPath(nn.Module): def __init__(self): super(SketchyPath, self).__init__() self.encoder = CNNEncoder() self.decoder = GenerativeDecoder() self.path_renderer = PathRenderer() sketchypath
def forward(self, x): x = torch.relu(self.conv1(x)) x = torch.relu(self.conv2(x)) x = x.view(-1, 128*28*28) x = torch.relu(self.fc1(x)) return x def forward(self, x): z = self