Han-Yi Wang henrywang1

Install nvidia driver

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-396 nvidia-modprobe
sudo reboot

	from __future__ import print_function
	import argparse
	import torch
	import torch.nn as nn
	import torch.nn.functional as F
	import torch.optim as optim
	from torchvision import datasets, transforms

	class Net(nn.Module):
	def __init__(self, share_block):

	# Push subtree in local branch to heroku
	git push heroku `git subtree split --prefix project-name local-branch-name`:master --force