From 71ab0cfcfc9a21be82fd7a9e631848a282e99fe9 Mon Sep 17 00:00:00 2001
From: channingss <chen_linghchi@163.com>
Date: Thu, 5 Sep 2019 16:54:38 +0800
Subject: [PATCH] fix bug for import torch

---
 x2paddle/convert.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/x2paddle/convert.py b/x2paddle/convert.py
index 7a522b7..000453f 100644
--- a/x2paddle/convert.py
+++ b/x2paddle/convert.py
@@ -190,8 +190,6 @@ def main():
             return
     except:
         print("paddlepaddle not installed, use \"pip install paddlepaddle\"")
-    assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)"
-    assert args.save_dir is not None, "--save_dir is not defined"
 
     if args.framework == "tensorflow":
         assert args.model is not None, "--model should be defined while translating tensorflow model"
-- 
GitLab