Non-trainable params:
8,537,680,896
(31.81 GB)