MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / _make_conv

Method _make_conv

sat/sgm/modules/autoencoding/temporal_ae.py:316–320  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

314 return super()._make_attn()
315
316 def _make_conv(self) -> Callable:
317 if self.time_mode != "attn-only":
318 return partialclass(AE3DConv, video_kernel_size=self.video_kernel_size)
319 else:
320 return Conv2DWrapper
321
322 def _make_resblock(self) -> Callable:
323 if self.time_mode not in ["attn-only", "only-last-conv"]:

Callers

nothing calls this directly

Calls 1

partialclassFunction · 0.90

Tested by

no test coverage detected