run.sh: add restore from checkpoint and output directory options