Searched refs:weight_bias_init (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 81 def __init__(self, input_dim, output_dim, weight_bias_init, act_str, argument
84 weight_init, bias_init = weight_bias_init
142 self.weight_init, self.bias_init = config.weight_bias_init
143 self.weight_bias_init = config.weight_bias_init
160 self.weight_bias_init,
164 self.weight_bias_init,
168 self.weight_bias_init,
172 self.weight_bias_init,
176 self.weight_bias_init,
|
| D | config.py | 59 self.weight_bias_init = ['normal', 'normal']
90 self.weight_bias_init = ['normal', 'normal']
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 131 …def __init__(self, input_dim, output_dim, weight_bias_init, act_str, scale_coef=1.0, convert_dtype… argument
134 weight_init, bias_init = weight_bias_init
205 self.weight_bias_init = config.weight_bias_init
215 … self.dense_layer_1 = DenseLayer(self.all_dim_list[0], self.all_dim_list[1], self.weight_bias_init,
217 … self.dense_layer_2 = DenseLayer(self.all_dim_list[1], self.all_dim_list[2], self.weight_bias_init,
219 … self.dense_layer_3 = DenseLayer(self.all_dim_list[2], self.all_dim_list[3], self.weight_bias_init,
221 … self.dense_layer_4 = DenseLayer(self.all_dim_list[3], self.all_dim_list[4], self.weight_bias_init,
223 … self.dense_layer_5 = DenseLayer(self.all_dim_list[4], self.all_dim_list[5], self.weight_bias_init,
|
| D | config.py | 43 weight_bias_init = ['normal', 'normal'] variable in ModelConfig
|