torchtnt.utils.test_utils.get_pet_launch_config

torchtnt.utils.test_utils.get_pet_launch_config(nproc: int) → LaunchConfig

Initialize pet.LaunchConfig for single-node, multi-rank functions.

Parameters:nproc – The number of processes to launch. Returns:An instance of pet.LaunchConfig for single-node, multi-rank functions.