New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

There is a size mismatch due to this lines #22

Open

exponentialR opened this issue Jul 12, 2022 · 0 comments

exponentialR commented Jul 12, 2022

densenet.pytorch/densenet.py

Lines 113 to 114 in d1cd5e1

    
           out = self.dense3(out) 
        
           out = torch.squeeze(F.avg_pool2d(F.relu(self.bn1(out)), 8))

I solve the issue by changing the lines of codes to

out = self.dense3(out) out = self.relu(self.bn1(out)) out = F.avg_pool2d(out, 8) out = out.view(-1, self.nChannels)

where self.relu has been initialised as self.relu = nn.ReLU(inplace=True)

The text was updated successfully, but these errors were encountered:

exponentialR changed the title ~~There is an size mismatch due to this lines~~ There is a size mismatch due to this lines

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment