This commit is contained in:
patil-suraj 2022-07-01 12:54:47 +02:00
parent 0dbc4779c8
commit db5a05742e
1 changed files with 1 additions and 1 deletions

View File

@ -86,7 +86,7 @@ class Upsample(nn.Module):
self.use_conv = use_conv
self.dims = dims
self.use_conv_transpose = use_conv_transpose
name = self.name
self.name = self.name
if use_conv_transpose:
conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)