Non-trainable params:
396,544
(1.51 MB)