Fixed a typo

guydav · Nov 28, 2022 · 6f7d5ab · 6f7d5ab
1 parent 2e28982
commit 6f7d5ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simple_relational_reasoning/embeddings/models.py b/simple_relational_reasoning/embeddings/models.py
@@ -187,7 +187,7 @@ def build_model(model_name: str, device: str, pretrained: bool = True, saycam: t
             model.embedding_dim = MODEL_EMBEDDING_DIMENSIONS[(model_name, unpooled_output)]  # type: ignore
             model.fc = nn.Identity()  # type: ignore
             if unpooled_output:
-                mode.avgpool = nn.Identity()  # type: ignore
+                model.avgpool = nn.Identity()  # type: ignore
             model = model.to(device)
 
     if model is None: