upt

Author: declmal

.gitignore

@@ -57,4 +57,4 @@ docs/doctrees
 docs/doxygen_output
 
 # django
-python/mrt/frontend/db.sqlite3
+python/mrt/web/db.sqlite3

python/mrt/V3/evaluate.py

@@ -35,7 +35,7 @@
         "--iter-num", type=int, default=10).dest: (_pname, "ITER_NUM"),
 })
 
-def evaluate(cm_cfg, pass_cfg):
+def evaluate(cm_cfg, pass_cfg, logger=None):
     model_dir = cm_cfg.MODEL_DIR
     model_name = cm_cfg.MODEL_NAME
     verbosity = cm_cfg.VERBOSITY
@@ -45,7 +45,8 @@ def evaluate(cm_cfg, pass_cfg):
     batch = pass_cfg.BATCH
 
     model_prefix = get_model_prefix(model_dir, model_name)
-    logger = get_logger(verbosity)
+    if logger is None:
+        logger = get_logger(verbosity)
     conf_quant_file = model_prefix + ".quantize.conf"
     check_file_existance(conf_quant_file, logger=logger)
     conf_map = load_conf(conf_quant_file, logger=logger)

python/mrt/V3/mrt_compile.py

@@ -35,7 +35,7 @@
         default=default_device_ids).dest: (_cnode, "DEVICE_IDS"),
 })
 
-def mrt_compile(cm_cfg, pass_cfg):
+def mrt_compile(cm_cfg, pass_cfg, logger=None):
     model_dir = cm_cfg.MODEL_DIR
     model_name = cm_cfg.MODEL_NAME
     verbosity = cm_cfg.VERBOSITY
@@ -45,7 +45,8 @@ def mrt_compile(cm_cfg, pass_cfg):
     batch = pass_cfg.BATCH
 
     model_prefix = get_model_prefix(model_dir, model_name)
-    logger = get_logger(verbosity)
+    if logger is None:
+        logger = get_logger(verbosity)
     conf_quant_file = model_prefix + ".quantize.conf"
     check_file_existance(conf_quant_file, logger=logger)
     conf_map = load_conf(conf_quant_file, logger=logger)

python/mrt/V3/quantize.py

@@ -48,7 +48,7 @@
         "--oscale-maps", type=tuple, default=[]).dest: (_pname, "OSCALE_MAPS"),
 })
 
-def quantize(cm_cfg, pass_cfg):
+def quantize(cm_cfg, pass_cfg, logger=None):
     model_dir = cm_cfg.MODEL_DIR
     model_name = cm_cfg.MODEL_NAME
     verbosity = cm_cfg.VERBOSITY
@@ -65,7 +65,8 @@ def quantize(cm_cfg, pass_cfg):
     oscale_maps = {opn1: opn2 for opn1, opn2 in pass_cfg.OSCALE_MAPS}
 
     model_prefix = get_model_prefix(model_dir, model_name)
-    logger = get_logger(verbosity)
+    if logger is None:
+        logger = get_logger(verbosity)
     conf_calib_file = model_prefix + ".calibrate.conf"
     check_file_existance(conf_calib_file, logger=logger)
     conf_map = load_conf(conf_calib_file, logger=logger)

python/mrt/web/web/urls.py

@@ -22,4 +22,5 @@
     path('calibrate/', views.mrt_calibrate_log),
     path('quantize/', views.mrt_quantize_log),
     path('evaluate/', views.mrt_evaluate_log),
+    path('compile/', views.mrt_compile_log),
 ]

python/mrt/web/web/views.py

@@ -12,7 +12,7 @@
 from mrt.V3.calibrate import calibrate
 from mrt.V3.quantize import quantize
 from mrt.V3.evaluate import evaluate
-# from mrt.V3.mrt_compile import mrt_compile
+from mrt.V3.mrt_compile import mrt_compile
 from .log import get_logger
 
 class Printer:
@@ -55,7 +55,7 @@ def start(self):
                 yield f'{item}<br>'
             except Empty:
                 pass
-        yield 'End'
+        yield '<br>***End***<br>'
         printer.clean(self.thread)
 
 mrt_web_tmp_dir = os.path.expanduser("~/.mrt_web")
@@ -96,3 +96,10 @@ def mrt_evaluate_log(request):
     logger = get_logger(cm_cfg.VERBOSITY, printer)
     streamer = Streamer(evaluate, (cm_cfg, pass_cfg, logger))
     return StreamingHttpResponse(streamer.start())
+
+def mrt_compile_log(request):
+    yaml_file = os.path.expanduser("~/mrt_yaml_root/alexnet.yaml")
+    cm_cfg, pass_cfg = get_cfg(yaml_file, "COMPILE")
+    logger = get_logger(cm_cfg.VERBOSITY, printer)
+    streamer = Streamer(mrt_compile, (cm_cfg, pass_cfg, logger))
+    return StreamingHttpResponse(streamer.start())