Trainable params:
803,329
(3.06 MB)