VanillaBlockΒΆ

Vanilla Convolution Block.

Abstract Signature:

VanillaBlock(input_dim: int, output_dim: int, kernel_size: int = 3)

PyTorch

API: torch.nn.Conv2d
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.VanillaBlock
Strategy: Direct Mapping