Train Neural Networks without Draining your Pocket: Profile your PyTorch Models

That fancy new model architecture might just be too fancy for your dataset and hardware — how about getting it profiled using PyTorch…