Change log for KE eval, 2

sayoojbk · Jul 6, 2019 · d4185ea · d4185ea
1 parent 1d86020
commit d4185ea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/convert_models.py b/convert_models.py
@@ -996,10 +996,10 @@ def main():
         pip_packages += ["cupy-cuda92", "cupy-cuda100", "chainer"]
     if (args.src_fwk == "keras") or (args.dst_fwk == "keras"):
         packages += ["keras"]
-        pip_packages += ["keras", "keras-mxnet", "keras-applications", "keras-preprocessing"]
+        pip_packages += ["keras", "keras-mxnet", "mxnet-cu100"]
     if (args.src_fwk == "tensorflow") or (args.dst_fwk == "tensorflow"):
         packages += ["tensorflow-gpu"]
-        pip_packages += ["tensorflow-gpu", "tensorpack", "mxnet-cu100"]
+        pip_packages += ["tensorflow-gpu", "tensorpack"]
 
     _, log_file_exist = initialize_logging(
         logging_dir_path=args.save_dir,

diff --git a/eval_ke.py b/eval_ke.py
@@ -97,12 +97,12 @@ def parse_args():
     parser.add_argument(
         '--log-packages',
         type=str,
-        default='keras',
+        default='keras, mxnet',
         help='list of python packages for logging')
     parser.add_argument(
         '--log-pip-packages',
         type=str,
-        default='keras, keras-mxnet, mxnet',
+        default='keras, keras-mxnet, mxnet-cu100',
         help='list of pip packages for logging')
     args = parser.parse_args()
     return args