Trainable params: 160,289 (626.13 KB)