if __name__ == '__main__':
model = ShufflenetV2(num_classes=1, training=False)
x = tf.random.uniform((2, 224,224, 3))
for ind in range(10000):
y = model(x)
model.build((1,224,224,3))
model.summary()
-
Notifications
You must be signed in to change notification settings - Fork 2
Zhengtq/shufflenetv2-tensorflow2.0
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
shufflenet v2 tensorflow2.0 tf2,0 tf-keras
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published