jactorch.train.utils

Functions

decay_learning_rate(optimizer, decay)	Decay the learning rate of the optimizer by a factor of decay.
mark_freezed(model)	DEPRECATED(Jiayuan Mao): jactorch.train.utils.mark_freezed has been deprecated and will be removed by 07/16/2022; please use jactorch.graph.parameter.mark_freezed instead.
mark_unfreezed(model)	DEPRECATED(Jiayuan Mao): jactorch.train.utils.mark_unfreezed has been deprecated and will be removed by 07/16/2022; please use jactorch.graph.parameter.mark_unfreezed instead.
set_learning_rate(optimizer, lr)	Set the learning rate of the optimizer.

Functions

decay_learning_rate(optimizer, decay)

Decay the learning rate of the optimizer by a factor of decay.

Parameters:

decay (float)

mark_freezed(model)

DEPRECATED(Jiayuan Mao): jactorch.train.utils.mark_freezed has been deprecated and will be removed by 07/16/2022; please use jactorch.graph.parameter.mark_freezed instead.

mark_unfreezed(model)

DEPRECATED(Jiayuan Mao): jactorch.train.utils.mark_unfreezed has been deprecated and will be removed by 07/16/2022; please use jactorch.graph.parameter.mark_unfreezed instead.

set_learning_rate(optimizer, lr)

Set the learning rate of the optimizer.

Parameters:

lr (float)