Sparse matrix computation in pytorch

Soumya_Kundu · September 5, 2024, 3:25pm

Here’s a gist explaining this

a clear solution on how I can achieve this in pytorch

gist.github.com

https://gist.github.com/aymuos15/10c8348148d3de5954a6b4cc5cfebf2c

Pruning_nuances_in_pytorch.py

import torch
import torch.nn as nn
import torch.nn.utils.prune as prune
import numpy as np
import matplotlib.pyplot as plt

class SimpleCNN(nn.Module):
    def __init__(self, conv1_out=32, conv2_out=64):
        super(SimpleCNN, self).__init__()
        self.conv1 = nn.Conv2d(1, conv1_out, 3, 1)

This file has been truncated. show original