如何记录/捕获python终端执行命令(Command)的全部内容?

在某些应用场景中,例如机器学习,运行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脚本时的所有参数,非常适用于需要追踪和文档化命令行操作的场景。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值