Trainable params: 10,241 (40.00 KB)