fix: provide option to use native process launcher

Author: Satish Pasumarthi

src/sagemaker_pytorch_container/training.py

@@ -81,6 +81,13 @@ def train(training_environment):
             runner_type = runner.PyTorchXLARunnerType
             logger.info('Invoking PT-XLA Runner')
     logger.info('Invoking user training script.')
+
+    # get capture_error from framework parameters
+    capture_error = True
+    if training_environment.additional_framework_parameters.get("sagemaker_toolkit_native_launcher_enabled"):
+        capture_error = False
+        logger.info(f'capture_error is {capture_error}. Default is True')
+
     try:
         entry_point.run(uri=training_environment.module_dir,
                         user_entry_point=training_environment.user_entry_point,