Skip to content

Commit e890867

Browse files
RdoubleAfacebook-github-bot
authored andcommitted
Rename attribute in SamePadConvTranspose to better match checkpoint (#207)
Summary: Simple change in `SamePadconvTranspose3d` from `self.conv` to `self.convt` for an easier mapping to MUGEN's checkpoint for VQVAE. Pull Request resolved: #207 Test Plan: `pytest test -vv` Reviewed By: langong347 Differential Revision: D38178382 Pulled By: RdoubleA fbshipit-source-id: 928ac24b946c42ce8404b9abbc985c10b9ceb65f
1 parent d710d46 commit e890867

File tree

1 file changed

+3
-3
lines changed
  • torchmultimodal/modules/layers

1 file changed

+3
-3
lines changed

torchmultimodal/modules/layers/conv.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def __init__(
108108
use ConvTranspose3d directly for custom padding"
109109
)
110110

111-
self.conv = nn.ConvTranspose3d(
111+
self.convt = nn.ConvTranspose3d(
112112
in_channels, out_channels, kernel_size, stride=stride, bias=bias, **kwargs
113113
)
114114

@@ -118,10 +118,10 @@ def forward(self, x: Tensor) -> Tensor:
118118
self.pad_input = calculate_same_padding(
119119
self.kernel_size, self.stride, x.shape[2:]
120120
)
121-
self.conv.padding, self.conv.output_padding = calculate_transpose_padding(
121+
self.convt.padding, self.convt.output_padding = calculate_transpose_padding(
122122
self.kernel_size, self.stride, x.shape[2:], self.pad_input
123123
)
124-
return self.conv(F.pad(x, self.pad_input))
124+
return self.convt(F.pad(x, self.pad_input))
125125

126126

127127
def calculate_same_padding(

0 commit comments

Comments
 (0)