Trainable params: 1,625,608 (6.20 MB)