Home
last modified time | relevance | path

Searched refs:loss_weights (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dfeature_columns_integration_test.py159 loss_weights = [1., 0.5]
164 loss_weights=loss_weights)
188 loss_weights = [1., 0.5]
193 loss_weights=loss_weights)
Dtraining_test.py117 loss_weights = [1., 2.]
118 model.compile(optimizer='adam', loss='mse', loss_weights=loss_weights)
124 loss_weights = {'dense_1': 1., 'dense_2': 2.}
125 model.compile(optimizer='adam', loss='mse', loss_weights=loss_weights)
168 model.compile(optimizer='adam', loss='mse', loss_weights=[1., 2.])
177 optimizer='adam', loss='mse', loss_weights={'unknown_output': 1.})
197 loss_weights = [1., 0.5]
202 loss_weights=loss_weights,
333 loss_weights = {'dense': 1., 'dropout': 0.5}
338 model.compile(optimizer, loss, metrics=metrics, loss_weights=loss_weights,
[all …]
Dtraining_utils.py1092 def prepare_loss_weights(output_names, loss_weights=None): argument
1112 if loss_weights is None:
1114 elif isinstance(loss_weights, dict):
1115 for name in loss_weights:
1120 weights_list = [loss_weights.get(name, 1.) for name in output_names]
1121 elif isinstance(loss_weights, list):
1122 if len(loss_weights) != len(output_names):
1127 str(loss_weights))
1128 weights_list = loss_weights
1131 str(loss_weights) + ' - expected a list of dicts.')
Dtraining_eager_test.py83 loss_weights = [1., 0.5]
89 loss_weights=loss_weights,
Ddistributed_training_utils.py662 loss_weights=model.loss_weights,
714 loss_weights=model.loss_weights,
Dtraining.py158 loss_weights=None, argument
286 self.loss_weights = loss_weights
334 self.output_names, loss_weights)
2471 loss_weights=self.loss_weights,
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format.py132 'loss_weights': model.loss_weights,
238 loss_weights = training_config['loss_weights']
246 loss_weights=loss_weights,
/external/tensorflow/tensorflow/python/keras/
Dmodels.py528 loss_weights=model.loss_weights,
/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_multi_worker_test.py136 loss_weights=model.loss_weights,
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py1032 loss_weights=self.model.loss_weights,
1425 self._cpu_model.loss_weights,
1451 loss_weights=None, argument
1465 metrics_module.clone_metrics(metrics), loss_weights, sample_weight_mode,
1469 super(KerasTPUModel, self).compile(optimizer, loss, metrics, loss_weights,
2250 model.loss_weights,
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead.py467 def multi_head(heads, loss_weights=None): argument
481 if loss_weights:
482 if len(loss_weights) != len(heads):
486 if loss_weights:
487 if len(losses) != len(loss_weights):
490 for loss, weight in zip(losses, loss_weights):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.models.-model.pbtxt154 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.-model.pbtxt154 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.-sequential.pbtxt159 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.layers.-linear-model.pbtxt159 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.models.-sequential.pbtxt159 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.models.-model.pbtxt154 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.-model.pbtxt154 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.-sequential.pbtxt159 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
Dtensorflow.keras.models.-sequential.pbtxt159 …argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'sample_weight_…
/external/tensorflow/tensorflow/lite/python/
Dlite_test.py1227 loss_weights=[1., 0.5])