actually fix the typo
This commit is contained in:
parent
db5a05742e
commit
60a981343e
|
@ -86,7 +86,7 @@ class Upsample(nn.Module):
|
||||||
self.use_conv = use_conv
|
self.use_conv = use_conv
|
||||||
self.dims = dims
|
self.dims = dims
|
||||||
self.use_conv_transpose = use_conv_transpose
|
self.use_conv_transpose = use_conv_transpose
|
||||||
self.name = self.name
|
self.name = name
|
||||||
|
|
||||||
if use_conv_transpose:
|
if use_conv_transpose:
|
||||||
conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)
|
conv = conv_transpose_nd(dims, channels, self.out_channels, 4, 2, 1)
|
||||||
|
|
Loading…
Reference in New Issue