BrandonLMorris

#!/usr/bin/env python3
"""
Brandon L Morris

Adapted from work by E. Culurciello
"""

from absl import app, flags
import itertools
from random import sample, randint, random

BrandonLMorris

from absl import app, flags
import torch
from torch import nn
import torch.nn.functional as F
from torchvision.datasets import FashionMNIST
from torchvision.transforms import Compose, ToTensor
from torch.utils.data import DataLoader

FLAGS = flags.FLAGS
dev = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

BrandonLMorris

FROM ubuntu:14.04

# Install Python Setuptools
RUN apt-get install -y python-setuptools

# Install pip
RUN easy_install pip

# Add and install Python Modules
ADD requirements.txt /src/requirements.txt