Update learn-embedding.sh