diff --git a/im2mesh/encoder/conv.py b/im2mesh/encoder/conv.py index 1668462..b394639 100644 --- a/im2mesh/encoder/conv.py +++ b/im2mesh/encoder/conv.py @@ -140,7 +140,7 @@ def __init__(self, c_dim, normalize=True, use_linear=True): super().__init__() self.normalize = normalize self.use_linear = use_linear - self.features = models.resnet50(pretrained=True) + self.features = models.resnet101(pretrained=True) self.features.fc = nn.Sequential() if use_linear: self.fc = nn.Linear(2048, c_dim)