Skip to content
Snippets Groups Projects
Commit 7ebd5ae4 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Changed to logging

parent f4c8b539
No related branches found
No related tags found
1 merge request!17Mccn extended
Pipeline #26566 failed
......@@ -17,8 +17,8 @@ from .utils import MaxFeatureMap
from .utils import group
from .utils import resblock
import bob.core
logger = bob.core.log.setup("bob.learn.pytorch")
import logging
logger = logging.getLogger("bob.learn.pytorch")
......@@ -51,6 +51,8 @@ class MCCNN(nn.Module):
num_channels: int
The number of channels present in the input
verbosity_level: int
Verbosity level.
"""
super(MCCNN, self).__init__()
......@@ -59,7 +61,7 @@ class MCCNN(nn.Module):
self.lcnn_layers=['conv1','block1','group1','block2', 'group2','block3','group3','block4','group4','fc']
bob.core.log.set_verbosity_level(logger, verbosity_level)
logger.setLevel(logger, verbosity_level)
self.pool1 = nn.MaxPool2d(kernel_size=2, stride=2, ceil_mode=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment