Skip to content

Commit

Permalink
🤖 Lint code
Browse files Browse the repository at this point in the history
  • Loading branch information
ivy-branch committed May 26, 2023
1 parent 99ac1b0 commit 9eef35e
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
8 changes: 6 additions & 2 deletions ivy_models/UNET/UNET.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
import ivy
ivy.set_backend('tensorflow')

ivy.set_backend("tensorflow")


def double_conv(in_c, out_c):
conv = ivy.Sequential(
Expand All @@ -10,6 +12,7 @@ def double_conv(in_c, out_c):
)
return conv


class UNet(ivy.Module):
def __init__(self):
super(UNet, self).__init__()
Expand All @@ -34,6 +37,7 @@ def _forward(self, image):
x9 = self.down_conv_5(x8)
print(x9.shape)


image = ivy.random_normal(shape=(1, 572, 572, 1))
model = UNet()
print(model(image))
print(model(image))
2 changes: 1 addition & 1 deletion ivy_models/UNET/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
from . import UNET
from .UNET import *
from .UNET import *

0 comments on commit 9eef35e

Please sign in to comment.