Skip to content

Instantly share code, notes, and snippets.

@bigsnarfdude
bigsnarfdude / dae_pytorch_cuda.py
Created December 9, 2017 00:17
denoising autoencoder pytorch cuda
import os
import torch
from torch import nn
from torch.autograd import Variable
from torch.utils.data import DataLoader
from torchvision import transforms
from torchvision.datasets import MNIST
from torchvision.utils import save_image