set_trainability: Set Trainability of Layers in Model

Description Usage Arguments Value

Description

This function sets the trainable attribute of a subset of the layers in a model to TRUE or FALSE.

Usage

1
set_trainability(.model, layer_names = NULL, trainability = FALSE)

Arguments

layer_names

(character vector) Names of the layers to set their trainable attribute.

trainability

(logical vector) New values for the trainable attribute of the selected layers.

model

(DLmodel) The model built by create_model_from_config.

Value

changes the trainable attribute in the selected layers in the model, in place.


neuroimaginador/dl4ni documentation built on May 3, 2019, 5:47 p.m.