MCPcopy Create free account
hub / github.com/ChunmingHe/WS-SAM / forward

Method forward

lib/Modules.py:485–489  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

483 self.sigmoid = nn.Sigmoid()
484
485 def forward(self, x):
486 avg_out = self.fc2(self.relu1(self.fc1(self.avg_pool(x))))
487 max_out = self.fc2(self.relu1(self.fc1(self.max_pool(x))))
488 out = avg_out + max_out
489 return self.sigmoid(out)
490
491
492class ODE(nn.Module):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected