Visualize and understand GPU memory in PyTorch - Tech Sentiments