Remove GPU in experiment

JuliaReinforcementLearning · Sep 25, 2023 · 8de7c84 · 8de7c84
1 parent 66c37a7
commit 8de7c84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ReinforcementLearningExperiments/deps/experiments/experiments/DQN/JuliaRL_NFQ_CartPole.jl b/...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,