Export mindir model
(model_predict, config)
| 175 | |
| 176 | |
| 177 | def export_mindir(model_predict, config): |
| 178 | """Export mindir model""" |
| 179 | inputs_np = Tensor(np.ones(shape=(config.batch_size, config.seq_length)), mstype.int32) |
| 180 | current_index = Tensor(np.array([0]), mstype.int32) |
| 181 | |
| 182 | batch_valid_length = Tensor(np.array([0]), mstype.int32) |
| 183 | init_true = Tensor([True], mstype.bool_) |
| 184 | inputs_np_1 = Tensor(np.ones(shape=(config.batch_size, 1)), mstype.int32) |
| 185 | |
| 186 | model_predict.predict_network.add_flags_recursive(is_first_iteration=True) |
| 187 | export(model_predict.predict_network, inputs_np, current_index, |
| 188 | init_true, batch_valid_length, file_name='pangu_alpha_1024', file_format='MINDIR') |
| 189 | model_predict.predict_network.add_flags_recursive(is_first_iteration=False) |
| 190 | export(model_predict.predict_network, inputs_np_1, current_index, |
| 191 | init_true, batch_valid_length, file_name='pangu_alpha_1', file_format='MINDIR') |
| 192 | print("Export finished and now exit.") |
| 193 | |
| 194 | |
| 195 | def run_predict(model_predict, config, args_opt, rank): |