fix error with iqn (#163)

JuliaReinforcementLearning · Apr 4, 2021 · 98af458 · 98af458
1 parent 69b77b4
commit 98af458
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/algorithms/dqns/iqn.jl b/src/algorithms/dqns/iqn.jl
@@ -182,7 +182,7 @@ function RLBase.update!(learner::IQNLearner, batch::NamedTuple)
     end
 
     aₜ = argmax(avg_zₜ, dims = 1)
-    aₜ = aₜ .+ typeof(aₜ)(CartesianIndices((0, 0:N′-1, 0)))
+    aₜ = aₜ .+ typeof(aₜ)(CartesianIndices((0:0, 0:N′-1, 0:0)))
     qₜ = reshape(zₜ[aₜ], :, batch_size)
     target =
         reshape(r, 1, batch_size) .+