Searched refs:extend_with_decoupled_weight_decay (Results 1 – 2 of 2) sorted by relevance
218 def extend_with_decoupled_weight_decay(base_optimizer): function
173 adamw = weight_decay_optimizers.extend_with_decoupled_weight_decay(