shivambhosale commited on
Commit
c4cb6b9
1 Parent(s): 80bb308

Update UNet.py

Browse files
Files changed (1) hide show
  1. UNet.py +1 -1
UNet.py CHANGED
@@ -1,4 +1,4 @@
1
- height, width = 512, 512
2
  class Block(Module):
3
  def __init__(self, inChannels, outChannels):
4
  super().__init__()
 
1
+ height, width = 256, 256
2
  class Block(Module):
3
  def __init__(self, inChannels, outChannels):
4
  super().__init__()