脚本如下:
frommindspore.train.serialization importexportfrommindspore importTensorinput=np.random.uniform(0.0,1.0,size=[1,3,32,32]).astype(np.float32) model =my_model_class() export(model,Tensor(input),file_name='backbone_nas',file_format='AIR')
报错信息如下
[ERROR]GE_ADPT(112164,python3):2021-01-05-10:34:21.938.071 [mindspore/ccsrc/transform/graph_ir/convert.cc:101]FindAdapter]Can't find OpAdapter for 2021-01-05 10:34:21.938 ERROR Failed to run pipeline. 2021-01-05 10:34:21.940 ERROR Traceback (most recent call last): File "/home/liyong/codes/vega_1221/automl/vega/core/pipeline/pipeline.py", line 55, in run PipeStep().do() File "/home/liyong/codes/vega_1221/automl/vega/core/pipeline/nas_pipe_step.py", line 43, in do self._dispatch_trainer(res) File "/home/liyong/codes/vega_1221/automl/vega/core/pipeline/nas_pipe_step.py", line 58, in _dispatch_trainer self.master.run(trainer, evaluator) File "/home/liyong/codes/vega_1221/automl/vega/core/scheduler/local_master.py", line 48, in run worker.train_process() File "/home/liyong/codes/vega_1221/automl/zeus/trainer/trainer_base.py", line 133, in train_process self._train_loop() File "/home/liyong/codes/vega_1221/automl/zeus/trainer/trainer_base.py", line 288, in _train_loop self._train_epoch() File "/home/liyong/codes/vega_1221/automl/zeus/trainer/trainer_ms.py", line 57, in _train_epoch export(self.model, Tensor(input), file_name='backbone_nas', file_format='AIR') File "/root/archiconda3/envs/ci3.7/lib/python3.7/site-packages/mindspore/train/serialization.py", line 528, in export _export(net, file_name, file_format, *inputs) File "/root/archiconda3/envs/ci3.7/lib/python3.7/site-packages/mindspore/train/serialization.py", line 553, in _export graph_id, _ = _executor.compile(net, *inputs, phase=phase_name) File "/root/archiconda3/envs/ci3.7/lib/python3.7/site-packages/mindspore/common/api.py", line 473, in compile self._build_data_graph(obj, phase) File "/root/archiconda3/envs/ci3.7/lib/python3.7/site-packages/mindspore/common/api.py", line 370, in _build_data_graph self._executor.build_data_graph(obj.parameters_dict(), phase, obj.parameters_broadcast_dict()) RuntimeError: mindspore/ccsrc/transform/graph_ir/convert.cc:101 FindAdapter] Can't findOpAdapter for Process finished with exitcode 0
解答:
当前导出AIR不支持包含控制流语义的网络,类似于网络construct中存在 for、while、if的语法。