Non-trainable params: 2,614,341,888 (9.74 GB)