experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Small refactor to improve the readability of the Model class for those who use the code as documentation.

Browse Source 
General idea: most important methods come first, private utilities are moved to the bottom of the class.
Also use a single method for `_standardize_user_data` (previously split into 2 methods that did not reflect
two separate sets of actions).

PiperOrigin-RevId: 225094903
This commit is contained in:
Francois Chollet 2018-12-11 16:45:06 -08:00 committed by TensorFlower Gardener
parent a54fd6b713
commit e8c65fa77f
2 changed files with 968 additions and 1022 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1983
tensorflow/python/keras/engine/training.py
View File

File diff suppressed because it is too large Load Diff

7
tensorflow/python/keras/engine/training_distributed.py
View File

@ -570,11 +570,8 @@ def _get_input_from_iterator(iterator, model):
# Validate that all the elements in x and y are of the same type and shape. # Validate that all the elements in x and y are of the same type and shape.
# We can then pass the first element of x and y to `_standardize_weights` # We can then pass the first element of x and y to `_standardize_weights`
# below and be confident of the output. # below and be confident of the output.
x_values, y_values, sample_weights_values = distributed_training_utils.\ distributed_training_utils.validate_distributed_dataset_inputs(
validate_distributed_dataset_inputs(model._distribution_strategy, x, y, model._distribution_strategy, x, y, sample_weights)
sample_weights)
model._standardize_weights(x_values, y_values,
sample_weight=sample_weights_values)
return x, y, sample_weights return x, y, sample_weights