Pytorch

I used Captum to interpret the output of a MobileNetV2, which visualized the main regions in the input image that drove the model to generate its output.

I converted a PyTorch model to Core ML and ran it on an iPhone.

Recently I was working with PyTorch multi-GPU training and I came across a nightmare GPU memory problem. After some expensive trial and error, I finally found a solution for it.