- ~~add script for downloading datasets~~ - fix resnet - add notebooks - ~~training~~ - class activation map visualisation - ~~use gpu~~ - save model - tensorboard