瀏覽代碼

changed logger initialization according to the newer implementation

Dimitri Korsch 5 年之前
父節點
當前提交
4d021e5056
共有 1 個文件被更改,包括 7 次插入2 次删除
  1. 7 2
      cvfinetune/parser/base.py

+ 7 - 2
cvfinetune/parser/base.py

@@ -19,7 +19,8 @@ def default_factory(extra_list=[]):
 
 
 class FineTuneParser(GPUParser):
-	def init_logger(self, simple=False, logfile=None):
+
+	def _logging_config(self, simple=False):
 		if not self.has_logging: return
 		fmt = '{levelname:s} - [{asctime:s}] {filename:s}:{lineno:d} [{funcName:s}]: {message:s}'
 
@@ -27,7 +28,11 @@ class FineTuneParser(GPUParser):
 		handler0.addFilter(HostnameFilter())
 		fmt0 = "<{hostname:^10s}>: " + fmt
 
-		filename = logfile if logfile is not None else f"{platform.node()}.log"
+		if self._args.logfile is None:
+			filename = f"{platform.node()}.log"
+		else:
+			filename = self._args.logfile
+
 		self._file_handler = handler1 = logging.FileHandler(filename=filename, mode="w")
 
 		logger_config.init_logging_handlers([