在某些应用场景中,例如机器学习,运行Python脚本时常常需要附加多个参数。例如,执行下面的命令:
python finetune.py --lora_r 16 --num_epochs 10 --prompt_template_name alpaca_short
那么如果想把脚本名和参数都记录下来,可以执行以下代码:
import sys
def main():
# 捕获完整的命令行,包括脚本名称
full_command = ' '.join(sys.argv)
print(full_command)
return full_command
if __name__ == "__main__":
main()
此时,你再运行上面的运行代码,终端打印为:
finetune.py --lora_r 16 --num_epochs 10 --prompt_template_name alpaca_short
如果我们只需要它的参数,可以使用以下Python代码:
import sys
def main():
# 仅捕获并打印出命令行中的参数部分,不包括脚本名称
full_command = ' '.join(sys.argv[1:])
print(full_command)
return full_command
if __name__ == "__main__":
main()
运行改进后的代码,终端将仅显示参数:
--lora_r 16 --num_epochs 10 --prompt_template_name alpaca_short
这样的脚本让用户能够灵活记录和查看运行Python脚本时的所有参数,非常适用于需要追踪和文档化命令行操作的场景。