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