Skip to content

Commit

Permalink
Logging format fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
torzdf committed Sep 25, 2019
1 parent 174e695 commit f55f8fc
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
2 changes: 2 additions & 0 deletions lib/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@

from numpy import ndarray


class FaceswapLogger(logging.Logger):
""" Create custom logger with custom levels """
def __init__(self, name):
Expand Down Expand Up @@ -69,6 +70,7 @@ def reformat_args(args):
new_args.append(arg)
return tuple(new_args)


class RollingBuffer(collections.deque):
"""File-like that keeps a certain number of lines of text in memory."""
def write(self, buffer):
Expand Down
2 changes: 1 addition & 1 deletion lib/training_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -423,7 +423,7 @@ def initialize(self, training_size):
warp_lm_edge_anchors=edge_anchors,
warp_lm_grids=grids)
self.initialized = True
logger.debug("Initialized constants: %s", {k:str(v) if isinstance(v, np.ndarray) else v
logger.debug("Initialized constants: %s", {k: str(v) if isinstance(v, np.ndarray) else v
for k, v in self._constants.items()})

# <<< TARGET IMAGES >>> #
Expand Down

0 comments on commit f55f8fc

Please sign in to comment.