From 8de7c84d1653cd6f3416f809d75ece81e0b8477d Mon Sep 17 00:00:00 2001
From: Cas Bex <cas.bex@kuleuven.be>
Date: Mon, 25 Sep 2023 13:14:27 +0200
Subject: [PATCH] Remove GPU in experiment

---
 .../deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ReinforcementLearningExperiments/deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl b/src/ReinforcementLearningExperiments/deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl
index de486114d..ef2e8aeab 100644
--- a/src/ReinforcementLearningExperiments/deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl
+++ b/src/ReinforcementLearningExperiments/deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl
@@ -34,7 +34,7 @@ function RLCore.Experiment(
                         Dense(ns, 32, σ; init=glorot_uniform(rng)),
                         Dense(32, 32, relu; init=glorot_uniform(rng)),
                         Dense(32, na; init=glorot_uniform(rng)),
-                    )|>gpu,
+                    ),
                     optimiser=RMSProp(),
                 ),
                 loss_function=mse,