((better)) — 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

class SketchyPath(nn.Module): def __init__(self): super(SketchyPath, self).__init__() self.encoder = CNNEncoder() self.decoder = GenerativeDecoder() self.path_renderer = PathRenderer() class CNNEncoder(nn