Trainable params: 1,363,968 (5.20 MB)