smash.Net.set_trainable

Net.set_trainable(trainable)

Method which enables to train or freeze the weights and biases of the network’s layers.

Parameters:

trainablelist of bool

List of booleans with a length of the total number of the network’s layers.

Note

Dropout, activation, and scaling functions are non-parametric layers, meaning they do not have any learnable weights or biases. Therefore, it is not necessary to set these layers as trainable since they do not involve any weight updates during training.