* [docker] megatron: add TE to ngc dockerfile * fix fa * add py-spy * set val_before_train=False for megatron
