learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-11-01 12:01:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

__call__ -> forward

Browse Source
This commit is contained in:
Varuna Jayasiri
2021-08-19 15:51:41 +05:30
parent bac2bf85e2
commit 0a4b5b6822
5 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
labml_nn/rl/dqn/model.py
View File

@ -82,7 +82,7 @@ class Model(Module):
nn.Linear(in_features=256, out_features=4),
)
def __call__(self, obs: torch.Tensor):
def forward(self, obs: torch.Tensor):
# Convolution
h = self.conv(obs)
# Reshape for linear layers