Pytorch

Interpret PyTorch Models with Captum

29 February 2020·5 mins

Deep-Learning 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.

From PyTorch to Core ML

16 October 2019·Updated: 31 January 2025·4 mins

Deep-Learning Pytorch Ios

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

Killing PyTorch Multi-GPU Training the Safe Way

2 November 2018·Updated: 31 January 2025·14 mins

Deep-Learning Pytorch

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.