ziyu-zui / heart-arrhythmia-diagnosis-with-deep-learning Goto Github PK

firstly, thanks for the excellent work,
secondly, can you provide the data of 'MIT_BIH_data_5class.p'

I noticed that there is a variable 'self.weight' in Class CNN that appears to be unused from my perspective, cuz it is unused in forward function. I know it might actually play a crucial role. Could you please explain the purpose or significance of that variable?
... ...
self.fc = nn.Linear(1984, 200)
self.out = nn.Linear(200, 5)
self.weight = Parameter(torch.Tensor(2, 32))

def forward(self, x):
    x = self.conv1(x)
    x = self.conv2(x)
    x = x.view(x.size(0), -1)

    x = F.relu(self.fc(x))
    x = F.dropout(x, 0.2)

    output = self.out(x)
    return output, x
    ... ...