Commit fb1514cc authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Disable print of network layers

parent efb4467f
......@@ -54,7 +54,7 @@ class BaseModel(ABC):
"""Calculate losses, gradients, and update network weights; called in every training iteration"""
pass
def setup(self):
def setup(self, verbose=False):
"""Load and print networks; create schedulers.
"""
if self.configuration['load_checkpoint'] >= 0:
......@@ -77,7 +77,8 @@ class BaseModel(ABC):
for _ in range(last_checkpoint):
s.step()
self.print_networks()
if verbose:
self.print_networks()
def train(self):
"""Make models train mode during test time."""
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment