From 90c322ea0cbe3a11ce079954550e224fdb65b7d7 Mon Sep 17 00:00:00 2001
From: Phil <s8phsaue@stud.uni-saarland.de>
Date: Thu, 3 Oct 2024 13:09:11 +0200
Subject: [PATCH] Bugfix qnetwork
---
src/networks.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/networks.py b/src/networks.py
index fbd1072..d0fed82 100644
--- a/src/networks.py
+++ b/src/networks.py
@@ -42,7 +42,7 @@ class QNetwork(nn.Module):
self.linear1 = nn.Linear(num_inputs + num_actions, hidden_dim)
self.linear2 = nn.Linear(hidden_dim, hidden_dim)
self.linear3 = nn.Linear(hidden_dim, 1)
- self.last_activation = F.sigmoid if sigmoid_activation else nn.Identity
+ self.last_activation = F.sigmoid if sigmoid_activation else nn.Identity()
self.apply(weights_init_)
--
GitLab