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

small fix

parent 2d54d90e
No related branches found
No related tags found
1 merge request!20Mccn trainer
Pipeline #26726 passed
......@@ -4,7 +4,7 @@ from torch.autograd import Variable
import torchvision.transforms as transforms
from bob.learn.pytorch.architectures import MCCNN
from bob.learn.pytorch.architectures import MCCNNv2
from bob.bio.base.extractor import Extractor
import logging
......@@ -43,7 +43,7 @@ class MCCNNv2Extractor(Extractor):
# model
self.transforms = transforms
self.network = MCCNN(num_channels=num_channels_used, adapted_layers=adapted_layers)
self.network = MCCNNv2(num_channels=num_channels_used, adapted_layers=adapted_layers)
logger.debug('Initiliazed model with lightCNN weights')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment