Table 4: Rot predictor.
Layer name Input size Output size
Linear 2048 2048
BN 2048 2048
ReLU 2048 2048
Linear 2048 512