Trainable params: 5,104,186,912 (10.39 GB)