diff --git a/python/paddle/fluid/layers/io.py b/python/paddle/fluid/layers/io.py
index c79f788de73e955234f49718a259e043bae307f7..f819625e08b181cd76f8dffd62eb62f287246387 100644
--- a/python/paddle/fluid/layers/io.py
+++ b/python/paddle/fluid/layers/io.py
@@ -1286,4 +1286,4 @@ def load(out, file_path, load_as_fp16=None):
     attrs = {"file_path": file_path}
     if load_as_fp16 is not None:
         attrs['load_as_fp16'] = load_as_fp16
-    helper.append_op(type="load", inputs={}, output={"Out": out}, args=attrs)
+    helper.append_op(type="load", inputs={}, output={"Out": out}, attrs=attrs)