From 3abf2e0fb6bb46a5d2948c24ebac3fa916125979 Mon Sep 17 00:00:00 2001
From: Junming Chen <72211694+Leoooo333@users.noreply.github.com>
Date: Sun, 16 Oct 2022 15:23:08 +0800
Subject: [PATCH] fix error: resnet101 typo

---
 im2mesh/encoder/conv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)