| /third_party/mindspore/mindspore/dataset/engine/ |
| D | validators.py | 41 _, param_dict = parse_user_args(method, *args, **kwargs)
48 dataset_dir = param_dict.get('dataset_dir')
51 validate_dataset_param_value(nreq_param_int, param_dict, int)
52 validate_dataset_param_value(nreq_param_bool, param_dict, bool)
53 validate_dataset_param_value(nreq_param_list, param_dict, list)
54 validate_dataset_param_value(nreq_param_dict, param_dict, dict)
55 check_sampler_shuffle_shard_options(param_dict)
57 cache = param_dict.get('cache')
70 _, param_dict = parse_user_args(method, *args, **kwargs)
75 dataset_dir = param_dict.get('dataset_dir')
[all …]
|
| /third_party/mindspore/mindspore/compression/export/ |
| D | quant_export.py | 131 param_dict=None): argument
137 self.core_op.add_prim_attr("filter_maxq", Tensor(param_dict["filter_maxq"]))
138 self.core_op.add_prim_attr("filter_minq", Tensor(param_dict["filter_minq"]))
139 if param_dict["output_maxq"] is not None:
140 self.core_op.add_prim_attr("output_maxq", Tensor(param_dict["output_maxq"]))
141 self.core_op.add_prim_attr("output_minq", Tensor(param_dict["output_minq"]))
142 self.core_op.add_prim_attr("symmetric", Tensor(param_dict["symmetric"]))
146 self.core_op.add_prim_attr("weight_num_bits", Tensor(param_dict["weight_num_bits"]))
147 self.core_op.add_prim_attr("weight_narrow_range", Tensor(param_dict["weight_narrow_range"]))
148 if param_dict["input_narrow_range"] is not None:
[all …]
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | utils.py | 32 def save_params(network, param_dict=None): argument
33 if param_dict is None:
37 if param.name in param_dict:
38 param_dict[param.name] = copy.deepcopy(param)
42 def restore_params(network, param_dict, init_adam=True): argument
50 elif param.name in param_dict:
51 param.set_data(param_dict[param.name])
53 if param.name in param_dict:
54 param.set_data(param_dict[param.name])
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | util.py | 59 param_dict = load_checkpoint(ckpt_path)
71 if darknet_weight in param_dict:
72 cell.weight.set_data(param_dict[darknet_weight].data)
76 if darknet_bias in param_dict:
77 cell.bias.set_data(param_dict[darknet_bias].data)
86 if darknet_moving_mean in param_dict:
87 cell.moving_mean.set_data(param_dict[darknet_moving_mean].data)
91 if darknet_moving_variance in param_dict:
92 cell.moving_variance.set_data(param_dict[darknet_moving_variance].data)
96 if darknet_gamma in param_dict:
[all …]
|
| D | initializer.py | 190 param_dict = load_checkpoint(args.resume_yolov3)
192 for key, values in param_dict.items():
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | test_fl_fasterrcnn.py | 157 param_dict = load_checkpoint(load_path)
165 for oldkey in list(param_dict.keys()):
167 data = param_dict.pop(oldkey)
169 param_dict[newkey] = data
174 param_dict[newkey] = param_dict.pop(oldkey)
176 for item in list(param_dict.keys()):
178 param_dict.pop(item)
180 for key, value in param_dict.items():
182 param_dict[key] = Parameter(tensor, key)
183 load_param_into_net(net, param_dict)
|
| /third_party/mindspore/mindspore/dataset/core/ |
| D | validator_helpers.py | 559 def check_sampler_shuffle_shard_options(param_dict): argument
568 shuffle, sampler = param_dict.get('shuffle'), param_dict.get('sampler')
569 num_shards, shard_id = param_dict.get('num_shards'), param_dict.get('shard_id')
570 num_samples = param_dict.get('num_samples')
590 def check_padding_options(param_dict): argument
601 columns_list = param_dict.get('columns_list')
602 padded_sample, num_padded = param_dict.get('padded_sample'), param_dict.get('num_padded')
642 def validate_dataset_param_value(param_list, param_dict, param_type): argument
651 if param_dict.get(param_name) is not None:
653 check_num_parallel_workers(param_dict.get(param_name))
[all …]
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_parallel_optimizer.py | 113 param_dict = train_network.parameter_layout_dict
115 assert not param_dict["weight1"][5]
116 assert param_dict["weight2"][5].startswith("4")
128 param_dict = train_network.parameter_layout_dict
130 assert not param_dict["weight1"][5]
131 assert not param_dict["weight2"][5]
139 param_dict = train_network.parameter_layout_dict
141 assert param_dict["weight1"][5].startswith("2")
142 assert param_dict["weight2"][5].startswith("2")
|
| /third_party/mindspore/scripts/ |
| D | update_onnx_weight.py | 35 param_dict = load_checkpoint(ckpt_file)
39 if not item.name in param_dict:
42 weight = param_dict[item.name].data.asnumpy()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | callbacks_ge.cc | 103 py::dict param_dict; in CheckpointSaveCallback() local
104 param_dict["name"] = name; in CheckpointSaveCallback()
105 param_dict["data"] = tensor_ptr; in CheckpointSaveCallback()
106 parameter_list.append(param_dict); in CheckpointSaveCallback()
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | test_lenet_quant.py | 50 param_dict = load_checkpoint(ckpt_path)
51 load_nonquant_param_into_quant_net(network, param_dict)
124 param_dict = load_checkpoint(ckpt_path)
125 not_load_param = load_param_into_net(network, param_dict)
|
| /third_party/mindspore/mindspore/nn/probability/bijector/ |
| D | bijector.py | 191 param_dict = self.parameters['param_dict']
193 for value in param_dict.values():
209 param_dict = self.parameters['param_dict']
210 for value in param_dict.values():
|
| /third_party/mindspore/tests/st/fl/albert/ |
| D | cloud_train.py | 164 param_dict = load_checkpoint(model_path)
165 if 'learning_rate' in param_dict:
166 del param_dict['learning_rate']
186 restore_params(server_network_train_cell, param_dict)
|
| D | cloud_eval.py | 75 param_dict = load_checkpoint(model_path)
76 restore_params(albert_model_cls, param_dict)
|
| /third_party/mindspore/tests/st/model_zoo_tests/deeplabv3/ |
| D | train_one_epoch_with_loss.py | 65 param_dict = load_checkpoint(checkpoint_path)
66 load_param_into_net(train_net, param_dict)
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 119 param_dict = load_checkpoint('checkpoint_lenet.ckpt') variable
120 load_param_into_net(network, param_dict)
|
| /third_party/mindspore/mindspore/nn/probability/distribution/ |
| D | distribution.py | 221 param_dict = self.parameters['param_dict']
222 for value in param_dict.values():
234 param_dict = self.parameters['param_dict']
235 for value in param_dict.values():
|
| /third_party/flutter/skia/third_party/externals/dawn/generator/ |
| D | generator_lib.py | 164 for param_dict in render.params_dicts:
165 params.update(param_dict)
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_serialize.py | 343 param_dict = load_checkpoint(ckpt_path, dec_key=key, dec_mode="AES-GCM")
344 load_param_into_net(net, param_dict)
347 param_dict = load_checkpoint(ckpt_path, dec_key=key, dec_mode="AES-GCM")
348 load_param_into_net(net, param_dict)
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet_cifar.py | 149 param_dict = load_checkpoint(args_opt.checkpoint_path) variable
150 load_param_into_net(net, param_dict)
|
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 97 param_dict = OrderedDict()
100 param_dict[name] = param
104 if param_dict:
105 inst_executor.run_init_graph(param_dict, init_phase)
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 890 param_dict = OrderedDict()
892 param_dict[param.name] = param
893 return param_dict
905 param_dict = OrderedDict()
908 param_dict[param.name] = param
909 if not param_dict:
911 return param_dict
|
| /third_party/skia/third_party/externals/dawn/generator/ |
| D | generator_lib.py | 183 for param_dict in render.params_dicts:
184 params.update(param_dict)
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 178 param_dict = load_checkpoint(load_checkpoint_path)
179 load_param_into_net(net_with_loss, param_dict)
|
| /third_party/mindspore/mindspore/train/ |
| D | serialization.py | 241 param_dict = {}
243 param_dict[param.name] = param
245 for (key, value) in param_dict.items():
1365 param_dict = {}
1424 param_dict[param.name] = split_param
1431 load_param_into_net(network, param_dict, strict_load=strict_load)
|