Skip to content

Commit adae09b

Browse files
committed
next fix for the last error
1 parent 6c86cdb commit adae09b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

R/training.R

+3
Original file line numberDiff line numberDiff line change
@@ -442,6 +442,9 @@ config_lr_scheduler <- function(type = c("lambda", "multiplicative", "reduce_on_
442442

443443
}else if (out$lr_scheduler == "step"){
444444
if("step_size" %in% names(mc)){
445+
if(class(mc$step_size)=="call"){
446+
mc$step_size <- eval(mc$step_size, envir=parent.frame(2L))
447+
}
445448
out$step_size <- mc$step_size
446449
if (verbose) cat(paste0("step_size: [", out$step_size, "]\n"))
447450
} else{

0 commit comments

Comments
 (0)