Skip to content
This repository was archived by the owner on May 6, 2021. It is now read-only.

Commit 98af458

Browse files
authored
fix error with iqn (#163)
1 parent 69b77b4 commit 98af458

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/algorithms/dqns/iqn.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ function RLBase.update!(learner::IQNLearner, batch::NamedTuple)
182182
end
183183

184184
aₜ = argmax(avg_zₜ, dims = 1)
185-
aₜ = aₜ .+ typeof(aₜ)(CartesianIndices((0, 0:N′-1, 0)))
185+
aₜ = aₜ .+ typeof(aₜ)(CartesianIndices((0:0, 0:N′-1, 0:0)))
186186
qₜ = reshape(zₜ[aₜ], :, batch_size)
187187
target =
188188
reshape(r, 1, batch_size) .+

0 commit comments

Comments
 (0)