From e12e06b9cdc82c8881dbb6b9679a9d1fd5f00cf9 Mon Sep 17 00:00:00 2001
From: lzhengning <lzhengning@gmail.com>
Date: Mon, 1 Mar 2021 14:52:33 +0800
Subject: [PATCH] doc: update load

---
 python/jittor/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py
index a37236a6..9d785ac9 100644
--- a/python/jittor/__init__.py
+++ b/python/jittor/__init__.py
@@ -932,6 +932,8 @@ class Module:
             >>> net.save('net.pkl')
             >>> net.load('net.pkl')
 
+        This method also support loading a state dict from a pytorch .pth file.
+
         .. note::
             当载入的参数与模型定义不一致时, jittor 会输出错误信息, 但是不会抛出异常.
             若载入参数出现模型定义中没有的参数名, 则会输出如下信息, 并忽略此参数: