can't get mnist tpu colab notebook to work with tpu
I ended up with following error when trying to run mnist_colab with tpu,
Exception in device=TPU:4: [Errno 2] No such file or directory: '/content/lightning_logs/version_0/checkpoints/epoch=0.ckpt' Traceback (most recent call last): File "/usr/local/lib/python3.6/dist-packages/torch_xla/distributed/xla_multiprocessing.py", line 119, in _start_fn fn(gindex, *args)

I've restarted the session after lightning import
I've set the num_tpu_cores=8
any ideas on what should be done?