WIP: Lightning

Working on a training mechanism using pytorch lightning

Still a WIP

bob/learn/pytorch/architectures/ResNet.py

@@ -281,6 +281,10 @@ def resnet18(pretrained=False, num_classes=1251, tp="cls", bn_dim=128):
         If True, returns a model pre-trained on ImageNet
 
     """
+    import ipdb
+
+    ipdb.set_trace()
+
     model = ResNet(
         BasicBlock, [2, 2, 2, 2], num_classes=num_classes, tp=tp, bn_dim=bn_dim
     )