why 'duration_discriminator_type' is redefined?

[5Hyeons]

vits2_pytorch/train_ms.py

Lines 172 to 188 in 1f4f379

	duration_discriminator_type = AVAILABLE_DURATION_DISCRIMINATOR_TYPES
	if duration_discriminator_type == "dur_disc_1":
	net_dur_disc = DurationDiscriminatorV1(
	hps.model.hidden_channels,
	hps.model.hidden_channels,
	3,
	0.1,
	gin_channels=hps.model.gin_channels if hps.data.n_speakers != 0 else 0,
	).cuda(rank)
	elif duration_discriminator_type == "dur_disc_2":
	net_dur_disc = DurationDiscriminatorV2(
	hps.model.hidden_channels,
	hps.model.hidden_channels,
	3,
	0.1,
	gin_channels=hps.model.gin_channels if hps.data.n_speakers != 0 else 0,
	).cuda(rank)

Due to this line, the net_dur_disc maintain None