Non-trainable params: 3 (16.00 B)