Fix qa config
> Approving, but is this change related to TLT at all?
> 
> Seems you are just polishing a bit functionality of a QA training NeMo script...

Should not affect tlt
this is rebased on https://github.com/NVIDIA/NeMo/pull/1721
please merge after that