bentrevett / pytorch-image-classification

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

t-sne resets the kernel.

erlebach opened this issue · comments

Great set of tutorials. However, it is so difficult to find tutorials that don't crash the system.
In the case of your MLP tutorial (the first one), all goes well until I execute t-sne. The Jupyter-lab (version 3) kernel resets.
Perhaps I do not have enough memory? It is possible. But a kernel reset is really a poor response since all my outputs must be recomputed, which takes time. So perhaps add checkpointing to your tutorial?

Here is an image of my screen when the kernel resets:
image