Trainable params: 2,928,640 (11.17 MB)