Index

A

afterany() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method) afternotok() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

afterok() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

B

BaseExperimentConfig (class in nntool.experiment) (class in nntool.experiment.config)

C

checkpoint() (nntool.slurm.Task method) (nntool.slurm.task.Task method) command() (nntool.slurm.PyTorchDistributedTask method) (nntool.slurm.Task method) (nntool.slurm.task.PyTorchDistributedTask method) (nntool.slurm.task.Task method)

configure() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

D

DistributedTaskConfig (class in nntool.slurm) (class in nntool.slurm.task)

E

export_bash() (nntool.slurm.DistributedTaskConfig method) (nntool.slurm.task.DistributedTaskConfig method)

G

get_current_time() (in module nntool.experiment) (in module nntool.experiment.utils) (in module nntool.utils) (nntool.experiment.BaseExperimentConfig method) (nntool.experiment.config.BaseExperimentConfig method)

get_output_path() (in module nntool.experiment) (in module nntool.experiment.utils) (in module nntool.utils) (nntool.experiment.BaseExperimentConfig method) (nntool.experiment.config.BaseExperimentConfig method)

I

init_wandb() (in module nntool.wandb) (in module nntool.wandb.config) is_configured() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

is_distributed() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

L

log() (nntool.slurm.Task method) (nntool.slurm.task.Task method)

M

map_array() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method) module nntool nntool.experiment nntool.experiment.config nntool.experiment.utils nntool.slurm nntool.slurm.accelerator nntool.slurm.accelerator.utils nntool.slurm.config nntool.slurm.function nntool.slurm.parser nntool.slurm.parser.parse nntool.slurm.task nntool.slurm.wrap nntool.utils nntool.wandb nntool.wandb.config

N

nntool module nntool.experiment module nntool.experiment.config module nntool.experiment.utils module nntool.slurm module nntool.slurm.accelerator module nntool.slurm.accelerator.utils module nntool.slurm.config module nntool.slurm.function module

nntool.slurm.parser module nntool.slurm.parser.parse module nntool.slurm.task module nntool.slurm.wrap module nntool.utils module nntool.wandb module nntool.wandb.config module nvidia_smi_gpu_memory_stats() (in module nntool.slurm.accelerator.utils) nvidia_smi_gpu_memory_stats_str() (in module nntool.slurm.accelerator.utils)

O

on_condition() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

P

PyTorchDistributedTask (class in nntool.slurm) (class in nntool.slurm.task)

R

read_toml_file() (in module nntool.experiment) (in module nntool.experiment.utils) (in module nntool.utils)

S

set_output_path() (nntool.slurm.config.SlurmConfig method) (nntool.slurm.SlurmConfig method) set_up_dist_env() (nntool.slurm.PyTorchDistributedTask method) (nntool.slurm.task.PyTorchDistributedTask method) set_up_stateful_fields() (nntool.experiment.BaseExperimentConfig method) (nntool.experiment.config.BaseExperimentConfig method) slurm_fn() (in module nntool.slurm) (in module nntool.slurm.wrap) slurm_function() (in module nntool.slurm) (in module nntool.slurm.wrap) slurm_launcher() (in module nntool.slurm) (in module nntool.slurm.wrap)

SlurmArgs (in module nntool.slurm) (in module nntool.slurm.config) SlurmConfig (class in nntool.slurm) (class in nntool.slurm.config) SlurmFunction (class in nntool.slurm) (class in nntool.slurm.function) start() (nntool.experiment.BaseExperimentConfig method) (nntool.experiment.config.BaseExperimentConfig method) submit() (nntool.slurm.function.SlurmFunction method) (nntool.slurm.SlurmFunction method)

T

Task (class in nntool.slurm) (class in nntool.slurm.task)

W

WandbConfig (class in nntool.wandb) (class in nntool.wandb.config)