Browse Source

fix 310 psenet

tags/v1.1.0
anzhengqi 5 years ago
parent
commit
0347d4e094
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/psenet/export.py

+ 1
- 1
model_zoo/official/cv/psenet/export.py View File

@@ -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)

Loading…
Cancel
Save