Wrappers for freeze() and unfreeze()

NAMESPACE

@@ -564,6 +564,7 @@ export(float)
 export(force_plot)
 export(foreground_acc)
 export(forget_mult_CPU)
+export(freeze)
 export(gan_critic)
 export(gan_loss_from_func)
 export(gauss_blur2d)
@@ -733,6 +734,7 @@ export(transformers)
 export(trunc_normal_)
 export(unet_config)
 export(unet_learner)
+export(unfreeze)
 export(uniform_blur2d)
 export(vgg11_bn)
 export(vgg13_bn)

R/freeze.R

@@ -0,0 +1,44 @@
+#' Freeze a model
+#'
+#' @param object A model 
+#'
+#' @return Nothing
+#' @export
+#'
+#' @examples
+#' \donrun{
+#' learnR %>% freeze()
+#' }
+freeze <- function(object){
+
+  args <- list()
+
+  do.call(object$freeze, args)
+  message("The model has been freezed")
+
+}
+
+#' Unfreeze a model
+#'
+#' @param object A model 
+#'
+#' @return Nothing
+#' @export
+#'
+#' @examples
+#' 
+#' \donrun{
+#' learnR %>% unfreeze()
+#' }
+unfreeze <- function(object){
+
+  args <- list()
+
+  do.call(object$unfreeze, args)
+  message("The model has been unfreezed")
+
+}
+
+
+
+