MCPcopy Create free account
hub / github.com/YeWR/EfficientZero / forward

Method forward

config/atari/model.py:174–184  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

172 )
173
174 def forward(self, x):
175 if self.downsample:
176 x = self.downsample_net(x)
177 else:
178 x = self.conv(x)
179 x = self.bn(x)
180 x = nn.functional.relu(x)
181
182 for block in self.resblocks:
183 x = block(x)
184 return x
185
186 def get_param_mean(self):
187 mean = []

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected