"Learn on the go" has a connotation of moving or using "downtime" effectively, whereas "learn on the fly" suggests a more immediate and pressing need to adapt.
Being able to learn on the go means you can transform any situation, whether it's traveling, waiting at a doctor's office, or standing in a queue, into a productive learning opportunity.