Torchsummary AttributeError: 'int' object has no attribute ...
discuss.pytorch.org › t › torchsummaryNov 07, 2020 · 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 ...