black formatted files before changes
This commit is contained in:
@@ -22,5 +22,5 @@ class BaseNet(nn.Module):
|
||||
"""Network summary."""
|
||||
net_parameters = filter(lambda p: p.requires_grad, self.parameters())
|
||||
params = sum([np.prod(p.size()) for p in net_parameters])
|
||||
self.logger.info('Trainable parameters: {}'.format(params))
|
||||
self.logger.info("Trainable parameters: {}".format(params))
|
||||
self.logger.info(self)
|
||||
|
||||
Reference in New Issue
Block a user