mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

conditioning masking is random

Browse Source
This commit is contained in:
James Betker 2022-07-01 21:43:30 -06:00
parent e06ee1b6f3
commit 286918c581
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/models/audio/music/tfdpc_v5.py
View File

@ -207,9 +207,10 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
if custom_conditioning_fetcher is not None:
cs, ce = custom_conditioning_fetcher(self.conditioning_encoder, time_emb)
else:
if self.conditioning_masking > 0:
if self.training and self.conditioning_masking > 0:
cond_op_len = x.shape[-1]
mask_len = int(cond_op_len * self.conditioning_masking)
mask_prop = random.random() * self.conditioning_masking
mask_len = int(cond_op_len * mask_prop)
if mask_len > 0:
start = random.randint(0, (cond_op_len-mask_len)) + cond_start
conditioning_input[:,:,start:(start+mask_len)] = 0