MConvBNReLU

class torchelie.nn.MConvBNReLU(in_ch: int, out_ch: int, ks: int, center=True)

Experimental: A packed block with Masked Conv-BN-ReLU

Parameters

• in_ch (int) – input channels

• out_ch (int) – output channels

• ks (int) – kernel size

• center (bool) – whether the masked conv has access to the central pixel or not

Returns

A packed block with MaskedConv-BN-ReLU as a CondSeq

Warning

MConvBNReLU() is experimental, and may change or be deleted soon if not already broken