建议在yaml配置文件中将常用的FLAGS加进去
Created by: nihuizhidao
PaddleDetection/tools/train.py里对FLAGS和cfg文件的读取顺序有点不太方便。。。 FLAGS需要在import paddle之前设置,读取配置文件是在import paddle之后,所以FLAGS不能在yaml配置文件里设置,导致还要改train.py代码,不能在配置文件里一步到位的修改涉及到FLAGS的参数。。。
是否可以考虑将某些常用的,比如FLAGS_selected_gpus等涉及到GPU操作的FLAG添加到yaml配置文件中,免去修改train.py的工作,更简洁点
另外,paddleDetection是默认会使用所有的GPU device么?我有两块,默认都用上了,实际上显存是一块就够够的了。。。
还有个问题想请教下,如果是多个PaddleDetection的不同程序/任务跑在同一块GPU上,使用auto_growth比较好,还是使用预分配显存(观察nvidia-smi得到某个程序的显存占用)的策略比较好呢?
感谢!