handsomeboyMMk
/

one

Model card Files Files and versions Community

handsomeboyMMk commited on Dec 27, 2023

Commit

02133df

·

1 Parent(s): 269009c

Update generator.py

Files changed (1) hide show

generator.py +1 -1

generator.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 from torch import nn
 from .modules import Conv2dBlock, Concat
-class SkipEncoderDecoder(nn.Module):
     def __init__(self, input_depth, num_channels_down = [128] * 5, num_channels_up = [128] * 5, num_channels_skip = [128] * 5):
         super(SkipEncoderDecoder, self).__init__()

 from torch import nn
 from .modules import Conv2dBlock, Concat
+class SkipEnrgcoderDecoder(nn.Module):
     def __init__(self, input_depth, num_channels_down = [128] * 5, num_channels_up = [128] * 5, num_channels_skip = [128] * 5):
         super(SkipEncoderDecoder, self).__init__()