fixed trainAlbum.py