iqn passes

JuliaReinforcementLearning · Mar 5, 2024 · 73ea25c · 73ea25c
1 parent 257fca0
commit 73ea25c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ReinforcementLearningZoo/src/algorithms/dqns/iqn.jl b/src/ReinforcementLearningZoo/src/algorithms/dqns/iqn.jl
@@ -132,5 +132,5 @@ function RLBase.optimise!(learner::IQNLearner, batch::NamedTuple)
         loss
     end
 
-    # RLBase.optimise!(A, gs)
+    RLBase.optimise!(A, gs[1])
 end