Update cnn.R

D-Maar · Oct 24, 2024 · c14e2ef · c14e2ef
1 parent a2ab8c5
commit c14e2ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/cnn.R b/R/cnn.R
@@ -129,7 +129,7 @@ cnn<-function (X = NULL, Y = NULL, architecture, loss = c("mse", "mae",
           early_stopping = NULL, lr_scheduler = NULL, custom_parameters = NULL,
           device = c("cpu", "cuda", "mps"), plot = TRUE, verbose = TRUE, train_data_loader=NULL, valid_data_loader=NULL)
 {
-  use_custom_dl<-!is.null(dataloader)
+  use_custom_dl<-!is.null(train_data_loader)
   if(use_custom_dl){
     sample_batch = train_data_loader$.iter()$.next()
     X<- as.array(sample_batch$X)