credit.pbs

credit.pbs#

Attributes#

logger

config_file

Functions#

launch_script(config_file, script_path[, launch])

Generates and optionally launches a PBS script for a single-node job.

launch_script_mpi(config_file, script_path[, launch, ...])

Generates and optionally launches a PBS script for a multi-node MPI job.

get_num_cpus()

Module Contents#

credit.pbs.logger#
credit.pbs.launch_script(config_file, script_path, launch=True)#

Generates and optionally launches a PBS script for a single-node job.

Parameters:

  • config_file (str) – Path to the YAML configuration file.

  • script_path (str) – Path to the script that will be executed by the PBS job.

  • launch (bool, optional) – If True, the PBS job will be submitted to the queue. Defaults to True.

credit.pbs.launch_script_mpi(config_file, script_path, launch=True, backend='nccl')#

Generates and optionally launches a PBS script for a multi-node MPI job.

Parameters:

  • config_file (str) – Path to the YAML configuration file.

  • script_path (str) – Path to the script that will be executed by the MPI job.

  • launch (bool, optional) – If True, the PBS job will be submitted to the queue. Defaults to True.

  • backend (str, optional) – Backend to be used for distributed training (e.g., ‘nccl’). Defaults to ‘nccl’.

credit.pbs.get_num_cpus()#
credit.pbs.config_file = '../config/vit2d.yml'#
Contents