fixed typo

D-Maar · Oct 24, 2024 · a2ab8c5 · a2ab8c5
1 parent 292c0ed
commit a2ab8c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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