From 9e4d9d1e5176f501edd9f4f3861affac487a5163 Mon Sep 17 00:00:00 2001
From: benedict-96 <benedikt.brantner@icloud.com>
Date: Fri, 20 Oct 2023 13:12:38 +0800
Subject: [PATCH] Fixed typo. Was calling the same routine recursively without
 doing anything.

---
 src/data_loader/data_loader.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/data_loader/data_loader.jl b/src/data_loader/data_loader.jl
index a92373606..1989f12c9 100644
--- 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"""