Fixed typo. Was calling the same routine recursively without doing an…

…ything.
JuliaGNI · Oct 20, 2023 · 9e4d9d1 · 9e4d9d1
1 parent 009a17c
commit 9e4d9d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/data_loader/data_loader.jl b/src/data_loader/data_loader.jl
@@ -100,11 +100,11 @@ function loss(model::Union{Chain, AbstractExplicitLayer}, ps::Union{Tuple, Named
 end
 
 function loss(model::Chain, ps::Tuple, dl::DataLoader{T, BT, Nothing}) where {T, BT<:AbstractArray{T, 3}} 
-    loss(model, ps, dl)
+    loss(model, ps, dl.input)
 end
 
 function loss(model::Chain, ps::Tuple, dl::DataLoader{T, BT, Nothing}) where {T, BT<:AbstractArray{T, 2}} 
-    loss(model, ps, dl)
+    loss(model, ps, dl.input)
 end
 
 @doc raw"""