update

app.py

@@ -3,8 +3,6 @@ os.system('python setup.py install --user')
 import argparse
 import csv
 import sys
-sys.path.insert(0, "/home/user/.local/lib/python3.8/site-packages")
-sys.path.append("/home/user/.local/lib/")
 sys.path.append("/home/user/.local/lib/python3.8/site-packages/diffvg-0.0.1-py3.8-linux-x86_64.egg")
 print(sys.path)
 from pathlib import Path
@@ -39,10 +37,10 @@ def run_cmd(command):
 # run_cmd("ls")
 # run_cmd("git submodule update --init --recursive")
 # run_cmd("python setup.py install --user")
-run_cmd("pip3 list")
-import pydiffvg
-
-print("Sccuessfuly import diffvg ")
+# run_cmd("pip3 list")
+# import pydiffvg
+#
+# print("Sccuessfuly import diffvg ")
 # run_cmd("pwd")
 # run_cmd("ls")
 # run_cmd("git submodule update --init --recursive")
@@ -96,9 +94,9 @@ def parse_args():
 
 
 
-
-def run_live(img, experiment_id):
-    main_func(img, experiment_id)
+cfg_arg = parse_args()
+def run_live(img, experiment_id, cfg_arg=cfg_arg):
+    main_func(img, experiment_id, cfg_arg=cfg_arg)
     return 0, 1
 
 

main.py

@@ -344,8 +344,7 @@ class linear_decay_lrlambda_f(object):
         lr = lr_s * (1-r) + lr_e * r
         return lr
 
-def main_func(target, experiment):
-    cfg_arg = parse_args()
+def main_func(target, experiment, cfg_arg):
     with open(cfg_arg.config, 'r') as f:
         cfg = yaml.load(f, Loader=yaml.FullLoader)
     cfg_default = edict(cfg['default'])