Created by: gfwm2013
目前FLAGS的设置需在import paddle之前通过设置environment的方式进行,如模型库中PaddleCV/ssd模型train.py的代码一样
现在可以通过set_flags
函数设置flags的值,无需要求在import paddle之前使用,具体用法:
flags = {'FLAGS_eager_delete_tensor_gb': 1.0}
fluid.set_flags(flags)
并且可以通过get_flags
接口来获取相关flag的值,具体用法:
flags = ['FLAGS_eager_delete_tensor_gb']
res = fluid.get_flags(flags)
print(res) # {'FLAGS_eager_delete_tensor_gb': 1.0}