From 8b9083963a5b964737c4c09acd241345971f99eb Mon Sep 17 00:00:00 2001 From: Skye Date: Thu, 5 Sep 2024 15:13:51 +0900 Subject: [PATCH] fix typo "temopral" --- lvdm/modules/networks/openaimodel3d.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lvdm/modules/networks/openaimodel3d.py b/lvdm/modules/networks/openaimodel3d.py index 49245da..bacbb36 100644 --- a/lvdm/modules/networks/openaimodel3d.py +++ b/lvdm/modules/networks/openaimodel3d.py @@ -187,7 +187,7 @@ def __init__( self.skip_connection = conv_nd(dims, channels, self.out_channels, 1) if self.use_temporal_conv: - self.temopral_conv = TemporalConvBlock( + self.temporal_conv = TemporalConvBlock( self.out_channels, self.out_channels, dropout=0.1, @@ -231,7 +231,7 @@ def _forward(self, x, emb, batch_size=None): if self.use_temporal_conv and batch_size: h = rearrange(h, '(b t) c h w -> b c t h w', b=batch_size) - h = self.temopral_conv(h) + h = self.temporal_conv(h) h = rearrange(h, 'b c t h w -> (b t) c h w') return h @@ -600,4 +600,4 @@ def forward(self, x, timesteps, context=None, features_adapter=None, fs=None, ** # reshape back to (b c t h w) y = rearrange(y, '(b t) c h w -> b c t h w', b=b) - return y \ No newline at end of file + return y