Imported from https://github.com/piergiaj/pytorch-i3d