Trainable params: 1,862,721 (7.11 MB)