Searched refs:is_weight_decay (Results 1 – 1 of 1) sorted by relevance
120 is_weight_decay=False): argument124 if is_weight_decay:144 is_weight_decay = False145 opt = Net(is_weight_decay)150 param, m, v, is_weight_decay)173 is_weight_decay = True174 opt = Net(is_weight_decay)179 param, m, v, is_weight_decay)203 is_weight_decay = False204 opt = SideEffectFusedAdamNet(is_weight_decay)[all …]