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