In this homework, we will train a convolutional network to classify images from SuperTuxKart.
This assignment should be solved individually. No collaboration, sharing of solutions, or exchange of models is allowed. Please, do not directly copy existing code from anywhere other than your previous solutions, or the previous master solution. We will check assignments for duplicates. See below for more details.
Starter code and dataset
The starter code for this assignment can be found here. The starter code contains several useful scripts:
bundle.pywill zip up your homework, ready for submission. Only submit zip files created by this bundling script
graderlocally grades your submission, works with both zip files and your homework directory.
The starter code also contains a
data directory where you’ll copy (or symlink) the SuperTuxKart classification dataset.
Unzip the data directly into the homework folder, replacing the existing data directory completely.
Make sure you see the following directories and files inside your main directory
homework grader bundle.py data data/train data/valid
You will run all scripts from inside this main directory.
CNN Model (10pts)
Similar to homework 1, your model should return a
torch.Tensor which represents the logits of the classes.
Use convolutions this time.
python -m grader homework -v to grade the first part.
- and all previous
In this part, we learn how to use
We created a dummy training procedure in
logging.py, and provided you with two
tb.SummaryWriter as logging utilities.
Use those summary writers to log the training loss at every iteration, the training accuracy at each epoch and the validation accuracy at each epoch.
Here is a simple example of how to use the
import torch.utils.tensorboard as tb logger = tb.SummaryWriter('cnn') logger.add_scalar('train/loss', t_loss, 0)
logging.py, you should not create your own
SummaryWriter, but rather use the one provided.
You can test your logger by calling
python -m homework.logging log, where
log is your favorite directory.
Then start up tensorboard:
tensoboard --logdir log.
python -m grader homework -v to grade the logging.
- and all previous
Training your CNN model (60pts)
Train your model and save it as
You can reuse some of the training functionality in
train.py from homework 1.
We highly recommend you incorporate the logging functionality from section 2 into your training routine.
Once you trained your model, you can optionally visualize your model’s prediction using
python -m homework.viz_prediction [DATASET_PATH].
After implementing everything, you can use
python -m grader homework to test your solutions against the validation grader locally.
Your model should achieve a 0.85 test accuracy to receive full points.
Note that we will use a testing dataset to grade the accuracy part of your model, so your local grades are not guaranteed to be your actual grades. (Don’t overfit!)
- and all previous
You can test your code using
python -m grader homework -v
This will run a subset of test cases we use during the actual testing. The point distributions will be the same, but we will use additional test cases. More importantly, we evaluate your model on the test set. The performance on the test grader may vary. Try not to overfit to the validation set too much.
Once you finished the assignment, create a submission bundle using
python bundle.py homework [YOUR UT ID]
and submit the zip file online. If you want to double-check that your zip file was properly created, you can grade it again
python -m grader [YOUR UT ID].zip
Running your assignment on google colab
You might need a GPU to train your models. You can get a free one on google colab. We provide you with a ipython notebook that can get you started on colab for each homework. Follow the instructions below to use it.
- Go to http://colab.research.google.com/.
- Sign in to your Google account.
- Select the upload tab then select the
- Go to Runtime -> Change runtime type.
GPUas the hardware accelerator.
- Follow the instructions on the homework notebook to upload code and data.
This assignment should be solved individually.
What interaction with classmates is allowed?
- Talking about high-level concepts and class material
- Talking about the general structure of the solution (e.g. You should use convolutions and ReLU layers)
- Looking at online solutions, and pytorch samples without directly copying or transcribing those solutions (rule of thumb, do not have your coding window and the other solution open at the same time). Always cite your sources in the code (put the full URL)!
- Using any of your submissions to prior homework
- Using the master solution to prior homework
- Using ipython notebooks from class
What interaction is not allowed?
- Exchange of code
- Exchange of architecture details
- Exchange of hyperparameters
- Directly (or slightly) modified code from online sources
- Any collaboration
- Putting your solution on a public repo (e.g. github). You will fail the assignment if someone copies your code.
Ways students failed in past years (do not do this):
Student A has a GPU, student B does not. Student B sends his solution to Student A to train 3 days before the assignment is due. Student A promises not to copy it but fails to complete the homework in time. In a last-minute attempt, Student A submits a slightly modified version of Student B’s solution. Result: Both students fail the assignment.
Student A struggles in class. Student B helps Student A and shows him/her his/her solution. Student A promises to not copy the solution but does it anyway. Result: Both students fail the assignment.
Student A sits behind Student B in class. Student B works on his homework, instead of paying attention. Student A sees Student B’s solution and copies it. Result: Both students fail the assignment.
Student A and B do not read the honor code and submit identical solutions for all homework. Result: Both students fail the class.
Installation and setup
Installing python 3
Installing the dependencies
Install all dependencies using
pip install -r requirements.txt
Note: On some systems, you might be required to use
pip3 instead of
pip for python 3.
If you’re using conda use
conda env create environment.yml
Manual installation of pytorch
Go to https://pytorch.org/get-started/locally/ then select the stable Pytorch build, your OS, package (pip if you installed python 3 directly, conda if you installed Anaconda), python version, cuda version. Run the provided command. Note that cuda is not required, you can select cuda = None if you don’t have a GPU or don’t want to do GPU training locally. We will provide instruction for doing remote GPU training on Google Colab for free.
Manual installation of the Python Imaging Library (PIL)
The easiest way to install the PIL is through
pip install -U Pillow
There are a few important considerations when using PIL.
First, make sure that your OS uses
libjpeg-turbo and not the slower
libjpeg (all modern Ubuntu versions do by default).
Second, if you’re frustrated with slow image transformations in PIL use
CC="cc -mavx2" pip install -U --force-reinstall Pillow-SIMD
CC="cc -mavx2" is only needed if your CPU supports AVX2 instructions.
pip will most likely complain a bit about missing dependencies.
Install them, either through
conda, or your favorite package manager (