Trainable params:
8,537,680,896
(15.90 GB)