localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Rename VisualTransformer -> VisionTransformer (#97)

Browse Source 
Fixes #94
This commit is contained in:
Haofan Wang 2021-07-19 11:41:49 +08:00 committed by GitHub
parent 504cf9e491
commit ea41722f9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clip/model.py
View File

@ -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,