Using the colab version speeds up things on GPU.
The mat-file can be obtained e.g. from Stanford.