From 5f0cc65f3b3b72f1d3534c4ccab0a331f751197d Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 5 Jun 2021 14:19:03 -0600 Subject: [PATCH] Register branched resnet properly --- codes/models/classifiers/cifar_resnet_branched.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/classifiers/cifar_resnet_branched.py b/codes/models/classifiers/cifar_resnet_branched.py index 03c6ac0b..42694438 100644 --- a/codes/models/classifiers/cifar_resnet_branched.py +++ b/codes/models/classifiers/cifar_resnet_branched.py @@ -143,7 +143,7 @@ class ResNet(nn.Module): return (tailouts[coarse_label] * torch.eye(n=bs).view(bs,bs,1)).sum(dim=1) @register_model -def register_cifar_resnet18(opt_net, opt): +def register_cifar_resnet18_branched(opt_net, opt): """ return a ResNet 18 object """ return ResNet(BasicBlock, [2, 2, 2, 2])