trainable params: 4,110,764 
all params: 562,951,596 
trainable%: 0.7302162440267778