| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers.py | 81 def __init__(self, weight_decay, **kwargs): argument
91 self._weight_decay = weight_decay
166 weight_decay = self._weight_decay
167 if callable(weight_decay):
168 weight_decay = weight_decay()
170 weight_decay, name="weight_decay")
270 def __init__(self, weight_decay, *args, **kwargs): argument
274 weight_decay, *args, **kwargs)
304 def __init__(self, weight_decay, learning_rate, momentum, argument
332 weight_decay, learning_rate=learning_rate, momentum=momentum,
[all …]
|
| D | lars_optimizer.py | 51 weight_decay=0.0001, argument
85 if weight_decay < 0.0:
86 raise ValueError("weight_decay should be positive: %s" % weight_decay)
91 self._weight_decay = weight_decay
|
| D | lars_optimizer_test.py | 50 weight_decay=wd_np,
97 weight_decay=wd_np,
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/ |
| D | densenet.py | 41 def __init__(self, num_filters, data_format, bottleneck, weight_decay=1e-4, argument
55 kernel_regularizer=l2(weight_decay))
66 kernel_regularizer=l2(weight_decay))
93 weight_decay=1e-4, dropout_rate=0): argument
104 kernel_regularizer=l2(weight_decay))
128 weight_decay=1e-4, dropout_rate=0): argument
138 weight_decay,
175 bottleneck=True, compression=0.5, weight_decay=1e-4, argument
186 self.weight_decay = weight_decay
232 self.weight_decay))
[all …]
|
| D | densenet_test.py | 45 weight_decay=1e-4, dropout_rate=0,
68 weight_decay=1e-4, dropout_rate=0,
91 weight_decay=1e-4, dropout_rate=0,
108 weight_decay = 1e-4
115 kernel_regularizer=tf.keras.regularizers.l2(weight_decay))
122 true_l2 = compute_true_l2(conv.variables, weight_decay)
128 loss_with_true_l2 = loss + compute_true_l2(conv.variables, weight_decay)
228 weight_decay=1e-4, dropout_rate=0,
277 weight_decay=1e-4, dropout_rate=0,
|
| D | densenet_graph_test.py | 62 weight_decay=1e-4, dropout_rate=0,
99 weight_decay=1e-4, dropout_rate=0,
129 weight_decay=1e-4, dropout_rate=0,
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| D | overfeat.py | 48 def overfeat_arg_scope(weight_decay=0.0005): argument
52 weights_regularizer=regularizers.l2_regularizer(weight_decay),
|
| D | alexnet.py | 52 def alexnet_v2_arg_scope(weight_decay=0.0005): argument
57 weights_regularizer=regularizers.l2_regularizer(weight_decay)):
|
| D | resnet_utils.py | 225 def resnet_arg_scope(weight_decay=0.0001, argument
257 weights_regularizer=regularizers.l2_regularizer(weight_decay),
|
| D | vgg.py | 56 def vgg_arg_scope(weight_decay=0.0005): argument
68 weights_regularizer=regularizers.l2_regularizer(weight_decay),
|
| D | inception_v1.py | 368 def inception_v1_arg_scope(weight_decay=0.00004, argument
408 weights_regularizer=regularizers.l2_regularizer(weight_decay)):
|
| D | inception_v2.py | 605 def inception_v2_arg_scope(weight_decay=0.00004, argument
636 weights_regularizer=regularizers.l2_regularizer(weight_decay)):
|
| D | inception_v3.py | 681 def inception_v3_arg_scope(weight_decay=0.00004, argument
722 weights_regularizer=regularizers.l2_regularizer(weight_decay)):
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| D | cifar10_pruning.py | 126 weight_decay = tf.multiply(tf.nn.l2_loss(var), wd, name='weight_loss')
127 tf.add_to_collection('losses', weight_decay)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 470 weight_decay = 0.01
473 regularizer = regularizers.l2_regularizer(weight_decay)
480 self.assertAlmostEqual(sess.run(wd), weight_decay * l2_loss.eval())
512 weight_decay = regularizers.l2_regularizer(0.01)
514 [layers_lib.convolution2d], weights_regularizer=weight_decay):
1727 weight_decay = regularizers.l2_regularizer(0.01)
1728 _layers.fully_connected(inputs, 32, weights_regularizer=weight_decay)
1759 weight_decay = regularizers.l2_regularizer(0.01)
1761 inputs, 32, weights_regularizer=weight_decay, scope='FC')
1766 inputs, 32, weights_regularizer=weight_decay, scope='FC', reuse=True)
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | revnet.py | 190 g + self.config.weight_decay * v if v.name.endswith("kernel:0") else g
|