Update test-train1k.sh