fix 310 psenet

2020-10-14 16:29:14 +08:00 · 2020-10-14 16:29:14 +08:00 · 0347d4e094
parent 11058ad0ef
commit 0347d4e094
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/psenet/export.py
+++ b/model_zoo/official/cv/psenet/export.py
@ -36,4 +36,4 @@ if __name__ == '__main__':

    input_arr = Tensor(np.random.uniform(0.0, 1.0, size=[1, 3, config.INFER_LONG_SIZE, config.INFER_LONG_SIZE]),
                       ms.float32)
-    export(net, input_arr, file_name=config.geir_filename)
+    export(net, input_arr, file_name=config.air_filename)