d6b9a5738af8c6dbe642b2464b4fec31c306aaa7,vocoder/train.py,,,#,32

Before Change

                                             setup_generator)

torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
torch.manual_seed(54321)
use_cuda = torch.cuda.is_available()
num_gpus = torch.cuda.device_count()
print(" > Using CUDA: ", use_cuda)
print(" > Number of GPUs: ", num_gpus)

After Change

                                             setup_generator)


use_cuda, num_gpus = setup_torch_training_env(True, True)


def setup_loader(ap, is_val=False, verbose=False):

Instances