def METHOD_NAME(self, batch_size):
    # create model
    config = FreeVCConfig()
    model = FreeVC(config).to(device)
    model.train()
    print(" > Num parameters for FreeVC model:%s" % (count_parameters(model)))
    _, _, mel, spec, spec_lengths, waveform = self._create_inputs(config, batch_size)
    wavlm_vec = model.extract_wavlm_features(waveform)
    wavlm_vec_lengths = torch.ones(batch_size, dtype=torch.long)
    y = model.forward(wavlm_vec, spec, None, mel, spec_lengths, wavlm_vec_lengths)
    # TODO: assert with training implementation