使用pytorch的时候,遇到一个作者的yaml读取问题,好久没用了,这里总结复习一下。
这是常用的yaml读写:
import yaml
##读取
with open("mmm.yaml", "r") as yaml_file:
yaml_obj = yaml.load(yaml_file.read())
yaml_paramer = yaml_obj["yaml_paramer "]
print(yaml_paramer )
# 写入
with open("./record/last_query_log_account.yaml", "w") as yaml_file:
yaml_obj["yaml_paramer"] = 0
yaml.dump(yaml_obj, yaml_file)