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

Correctly initializing the logit scale parameter

Browse Source 
adding numpy import
This commit is contained in:
Jong Wook Kim 2021-03-22 22:09:57 -04:00 committed by GitHub
parent 43c953e231
commit 290ac5cb15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
clip/model.py
View File

@ -1,6 +1,7 @@
from collections import OrderedDict from collections import OrderedDict
from typing import Tuple, Union from typing import Tuple, Union
import numpy as np
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from torch import nn from torch import nn