Yes.  The more you take your baby swimming and practice the skills you have been learning in class, the better your baby will progress.  Remember to take some toys and alternate exercise with play.  If you can get together with some other mums in your class to practice, this will help with exercises and make the sessions more fun too!  It is especially important to practice during the breaks for holidays.