WIP: Lightning

Working on a training mechanism using pytorch lightning

Still a WIP

bob/learn/pytorch/architectures/iresnet.py

@@ -183,7 +183,12 @@ class IResNet(nn.Module):
 def _iresnet(arch, block, layers, pretrained, progress, **kwargs):
     model = IResNet(block, layers, **kwargs)
     if pretrained:
-        raise ValueError()
+
+        map_location = (
+            torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+        )
+        state_dict = torch.load(pretrained, map_location=map_location)
+        model.load_state_dict(state_dict)
     return model