Trainable params:
407,050
(1.55 MB)