From a2ab8c5dd81282203b225698def749b6c477e9fb Mon Sep 17 00:00:00 2001
From: danielMaar <70383336+D-Maar@users.noreply.github.com>
Date: Thu, 24 Oct 2024 15:46:19 +0200
Subject: [PATCH] fixed typo

---
 R/cnn.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/R/cnn.R b/R/cnn.R
index 92423ab..8cb45d4 100644
--- a/R/cnn.R
+++ b/R/cnn.R
@@ -131,7 +131,7 @@ cnn<-function (X = NULL, Y = NULL, architecture, loss = c("mse", "mae",
 {
   use_custom_dl<-!is.null(dataloader)
   if(use_custom_dl){
-    sample_batch = data_loader$.iter()$.next()
+    sample_batch = train_data_loader$.iter()$.next()
     X<- as.array(sample_batch$X)
     Y<- c(as.matrix(sample_batch$Y))
     batchsize=2L