trainable 6,912 of 355,584 parameters (1.9%)
step 0  loss 6.920
step 50  loss 0.611
step 100  loss 0.248
step 150  loss 0.147
