rename perplexity->log perplexity
This commit is contained in:
parent
1d68624828
commit
27a9b1b750
|
@ -659,7 +659,7 @@ class GaussianDiffusion:
|
|||
img = model_driven_out + guidance_driven_out
|
||||
return img
|
||||
|
||||
def p_sample_loop_for_perplexity(
|
||||
def p_sample_loop_for_log_perplexity(
|
||||
self,
|
||||
model,
|
||||
truth,
|
||||
|
|
|
@ -137,8 +137,8 @@ class MusicDiffusionFid(evaluator.Evaluator):
|
|||
# s = q9.clamp(1, 9999999999)
|
||||
# x = x.clamp(-s, s) / s
|
||||
# return x
|
||||
perp = self.diffuser.p_sample_loop_for_perplexity(self.model, mel_norm,
|
||||
model_kwargs = {'truth_mel': mel_norm})
|
||||
perp = self.diffuser.p_sample_loop_for_log_perplexity(self.model, mel_norm,
|
||||
model_kwargs = {'truth_mel': mel_norm})
|
||||
|
||||
sampler = self.diffuser.ddim_sample_loop if self.ddim else self.diffuser.p_sample_loop
|
||||
gen_mel = sampler(self.model, mel_norm.shape, model_kwargs={'truth_mel': mel_norm})
|
||||
|
@ -317,7 +317,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
|
|||
self.local_modules[k] = mod.cpu()
|
||||
self.spec_decoder = self.spec_decoder.cpu()
|
||||
|
||||
return {"frechet_distance": frechet_distance, "perplexity": perplexity}
|
||||
return {"frechet_distance": frechet_distance, "log_perplexity": perplexity}
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
Loading…
Reference in New Issue
Block a user