Be careful that you need to be on pytorch master to have the latest grid_sample function when running this notebook. We specifically had the pytorch team change it to do the data augmentation as we wanted it (explained in this topic) so we asked them to support different kinds of padding_mode and interpolations, as well as make it more efficient (this last change has just been merged into master I believe).
2 Likes