From 24bc3e364bf09007cf46a8f6989da9a478629905 Mon Sep 17 00:00:00 2001
From: wangzeyangyi <tomzwang11@gmail.com>
Date: Tue, 13 Sep 2022 16:52:52 +0800
Subject: [PATCH] fix
---
official/cv/predrnn++/export.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/official/cv/predrnn++/export.py b/official/cv/predrnn++/export.py
index e5cc24385..220a8dfac 100644
--- a/official/cv/predrnn++/export.py
+++ b/official/cv/predrnn++/export.py
@@ -60,7 +60,7 @@ def run_export():
int(config.patch_size)**2*int(config.img_channel)))
input_arr = (Tensor(np.ones(shape), dtype=ms.float32), Tensor(mask_true, dtype=ms.float32))
- export(network, *input_arr, file_name=config.file_name, files_format=config.file_format)
+ export(network, *input_arr, file_name=config.file_name, file_format=config.file_format)
if __name__ == '__main__':
--
GitLab