luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

resnet 101 working

Browse Source
This commit is contained in:
luni
2023-02-21 22:30:32 +00:00
parent 8447b7c505
commit 9eb1183df4
4 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/resnet.py
+4 -4
View File
@@ -272,8 +272,8 @@ class ResNetTypeII(tf.keras.Model):
return x
def resnet_18(features, reversed=False, labels=None, is_cvae=False):
return ResNetTypeI(layer_params=[2, 2, 2, 2, 2], features=features, reversed=reversed, labels=labels, is_cvae=is_cvae)
def resnet_18(layer_params, features, reversed=False, labels=None, is_cvae=False):
return ResNetTypeI(layer_params=layer_params, features=features, reversed=reversed, labels=labels, is_cvae=is_cvae)
def resnet_34():
@@ -284,8 +284,8 @@ def resnet_50():
return ResNetTypeII(layer_params=[3, 4, 6, 3])
def resnet_101(features, reversed=False, labels=None, is_cvae=False):
return ResNetTypeII(layer_params=[3, 4, 23, 3, 3], features=features, reversed=reversed, labels=labels, is_cvae=is_cvae)
def resnet_101(layer_params, features, reversed=False, labels=None, is_cvae=False):
return ResNetTypeII(layer_params=layer_params, features=features, reversed=reversed, labels=labels, is_cvae=is_cvae)
def resnet_152():