| /external/pytorch/test/distributed/_composable/fully_shard/ |
| D | test_fully_shard_model_checkpoint.py | 155 load_model = TransformerWithSharedParams.init(
160 _zero_model(load_model, zero_buffers=True, summon_full=False)
162 load_model,
165 load_model.get_ignored_modules() if ignore_modules else []
169 FSDP.set_state_dict_type(load_model, StateDictType.FULL_STATE_DICT)
171 FSDP.set_state_dict_type(load_model, StateDictType.SHARDED_STATE_DICT)
172 load_model.load_state_dict(state_dict)
173 self._check_model_parity(load_model, save_model)
|
| /external/libopus/scripts/ |
| D | dump_rnn.py | 9 from keras.models import load_model
36 model = load_model("weights.hdf5", custom_objects={'binary_crossentrop2': binary_crossentrop2})
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | save.py | 155 def load_model(filepath, custom_objects=None, compile=True, options=None): # pylint: disable=redef… function
214 keras_deps.register_load_model_function(load_model)
|
| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_hybrid_shard.py | 163 load_model = fsdp_ctor(MyModel().cuda())
164 load_optim = torch.optim.AdamW(load_model.parameters())
165 with FSDP.state_dict_type(load_model, StateDictType.SHARDED_STATE_DICT):
166 load_model.load_state_dict(msd)
167 FSDP.optim_state_dict_to_load(load_model, load_optim, osd)
|
| /external/rnnoise/training/ |
| D | dump_rnn.py | 9 from keras.models import load_model
78 model = load_model(sys.argv[1], custom_objects={'msse': mean_squared_sqrt_error, 'mean_squared_sqrt…
|
| /external/libopus/training/ |
| D | rnn_dump.py | 11 from keras.models import load_model
|
| /external/pytorch/benchmarks/distributed/rpc/parameter_server/ |
| D | launcher.py | 133 model = load_model(args)
372 def load_model(args): function
|
| /external/pytorch/test/quantization/core/experimental/ |
| D | quantization_util.py | 87 def load_model(model_file): function
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 45 load_model = save.load_model variable
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distributed_table_test.py | 401 load_fn = keras_save.load_model
431 load_fn = keras_save.load_model
|
| /external/pytorch/benchmarks/dynamo/ |
| D | timm_models.py | 260 def load_model( member in TimmRunner
|
| D | torchbench.py | 221 def load_model( member in TorchBenchmarkRunner
|
| D | huggingface.py | 401 def load_model( member in HuggingfaceRunner
|
| D | common.py | 2231 yield self.load_model(
2305 device, name, model, example_inputs, _ = self.load_model(
4115 ) = runner.load_model(
4132 ) = runner.load_model(
4145 ) = runner.load_model(
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | README.md | 32 * [`tf.keras.models.load_model`](https://www.tensorflow.org/api_docs/python/tf/keras/models/load_mo…
|
| /external/pytorch/test/jit/ |
| D | test_cuda.py | 666 load_model = self.getExportImportCopy(script_model)
667 is_stream_s, a_load, b_load, c_load = load_model()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_test_util.py | 43 from tensorflow.python.saved_model import load as load_model unknown
878 m2 = load_model.load(saved_model_dir)
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_test.py | 2316 keras_model = keras.models.load_model(self._keras_file)
2346 keras_model = keras.models.load_model(
2468 keras_model = keras.models.load_model(self._keras_file)
2673 keras_model = keras.models.load_model(self._keras_file)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | extension_type_test.py | 1094 loaded_model = keras_save.load_model(path)
|
| /external/tensorflow/ |
| D | RELEASE.md | 6383 `tf.keras.models.load_model` instead.
6636 `tf.keras.models.load_model` instead.
6776 `tf.keras.models.load_model`. HDF5 files are still supported, and may be
|