Commit 60a98134 authored by patil-suraj's avatar patil-suraj
Browse files

actually fix the typo

parent db5a0574
...@@ -86,7 +86,7 @@ class Upsample(nn.Module): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment