nnsvs.train_util.setup

nnsvs.train_util.setup(config, device, collate_fn=<function collate_fn_default>)[source]

Setup for training

Parameters:

config (dict) – configuration for training
device (torch.device) – device to use for training
collate_fn (callable, optional) – collate function. Defaults to collate_fn_default.

Returns:

tuple containing model, optimizer, learning rate scheduler,: data loaders, tensorboard writer, logger, and scalers.

Return type: