Searched refs:training_config (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 176 … training_config = model._get_compile_args(user_metrics=False) # pylint: disable=protected-access
177 training_config.pop('optimizer', None) # Handled separately.
178 metadata['training_config'] = _serialize_nested_config(training_config)
204 def compile_args_from_training_config(training_config, custom_objects=None): argument
210 optimizer_config = training_config['optimizer_config']
215 loss_config = training_config.get('loss', None)
221 metrics_config = training_config.get('metrics', None)
227 weighted_metrics_config = training_config.get('weighted_metrics', None)
232 sample_weight_mode = training_config['sample_weight_mode'] if hasattr(
233 training_config, 'sample_weight_mode') else None
[all …]
|
| D | hdf5_format.py | 193 training_config = f.attrs.get('training_config')
194 if hasattr(training_config, 'decode'):
195 training_config = training_config.decode('utf-8')
196 if training_config is None:
200 training_config = json_utils.decode(training_config)
204 training_config, custom_objects))
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | load.py | 168 training_config = model._serialized_attributes['metadata'].get(
170 if training_config is not None:
172 training_config))
|
| /external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/ |
| D | saved_model.pbtxt | 8465 …hape\": {\"class_name\": \"__tuple__\", \"items\": [null, 214]}}}, \"training_config\": {\"loss\":…
|