Add test stage to setup to finish megatron init
If only trainer.test was being called then megatron was not finishing it's init.

This should fix this issue: https://github.com/NVIDIA/NeMo/issues/1700

