Hello everyone, I built a simple model and I want to know the number of its parameters using torchsummary, but I got an error: “AttributeError: ‘int’ object has no attribute 'numpy”
class LinearRegression(nn.Module):
def __init__(self, in_features: int, out_features: int, bias: bool = True):
        super().__init__()
        self.weights = nn.Parameter(torch.randn((in_features,out_features),requires_grad=True))

        self.bias = bias
        if bias:            
            self.bias_term = nn.Parameter(torch.rand((out_features),requires_grad=True))
    def forward(self, x):
        x =  x@ self.weights
        if self.bias:
            x +=  self.bias_term
        return x
from torchsummary import summary
linear_regression = LinearRegression(2, 1)
summary(model=linear_regression,input_size=(4,2),device='cpu')

Please tell me what am I doing wrong?