Skip to content

Instantly share code, notes, and snippets.

@reprompting
reprompting / 01_train.py
Last active May 13, 2026 21:13
sketch rnn (colab friendly)
# sketch-rnn for cats
# Implementing "A Neural Representation of Sketch Drawings"
import os, math, time, urllib.request
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
from tqdm.auto import tqdm