VedantDave/DB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding Files

Browse Source
This commit is contained in:
Vedant Dave 2023-05-29 13:34:41 +02:00
parent e45c6ab696
commit 061057fa11
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils.py
View File

@ -292,7 +292,7 @@ class GenerativeNetworkGaussian(tf.keras.Model):
return x
class ProjectionHead(tf.keras.Model):
class ProjectionHead(tf.keras.Model):
def __init__(self, name=None):
super(ProjectionHead, self).__init__(name=name)
self.dense1 = layers.Dense(256, activation=None)