Layer

Layer parameters

Input Channels

Output Channels

Subhead

Stride

Conv2D

2

16

(1,16)

8

BatchNorm2D

16

16

―

―

ReLU

―

―

―

―

Conv2D

16

16

(1,16)

8

BatchNorm2D

16

16

―

―

ReLU

―

―

―

―