MCPcopy Create free account
hub / github.com/TingsongYu/PyTorch_Tutorial / forward

Method forward

Code/4_viewer/2_visual_weights.py:21–28  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

19 self.fc3 = nn.Linear(84, 10)
20
21 def forward(self, x):
22 x = self.pool1(F.relu(self.conv1(x)))
23 x = self.pool2(F.relu(self.conv2(x)))
24 x = x.view(-1, 16 * 5 * 5)
25 x = F.relu(self.fc1(x))
26 x = F.relu(self.fc2(x))
27 x = self.fc3(x)
28 return x
29
30 # 定义权值初始化
31 def initialize_weights(self):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected