Merge pull request #310 from littletomatodonkey/dyg/fix_export

fix export model
pull/312/head
littletomatodonkey 2020-10-16 14:33:13 +08:00 committed by GitHub
commit fa24f82344
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions

View File

@ -13,6 +13,11 @@
# limitations under the License.
import argparse
import os
import sys
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.append(os.path.abspath(os.path.join(__dir__, '..')))
from ppcls.modeling import architectures
from ppcls.utils.save_load import load_dygraph_pretrain