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