From ea41722f9fe6bb81189f41ab6c306e77e930c2af Mon Sep 17 00:00:00 2001 From: Haofan Wang Date: Mon, 19 Jul 2021 11:41:49 +0800 Subject: [PATCH] Rename VisualTransformer -> VisionTransformer (#97) Fixes #94 --- clip/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clip/model.py b/clip/model.py index 422a34a..f2c95c4 100644 --- a/clip/model.py +++ b/clip/model.py @@ -199,7 +199,7 @@ class Transformer(nn.Module): return self.resblocks(x) -class VisualTransformer(nn.Module): +class VisionTransformer(nn.Module): def __init__(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, output_dim: int): super().__init__() self.input_resolution = input_resolution @@ -266,7 +266,7 @@ class CLIP(nn.Module): ) else: vision_heads = vision_width // 64 - self.visual = VisualTransformer( + self.visual = VisionTransformer( input_resolution=image_resolution, patch_size=vision_patch_size, width=vision_width,