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

revert fill_gaps back to old masking behavior

Browse Source
This commit is contained in:
James Betker 2022-05-08 00:10:19 -06:00
parent 58ed27d7a8
commit 7812c23c7a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/music_gen_fill_gaps.py
View File

@ -172,7 +172,7 @@ class MusicGenerator(nn.Module):
def do_masking(self, truth): def do_masking(self, truth):
b, c, s = truth.shape b, c, s = truth.shape
mask = torch.ones_like(truth) mask = torch.ones_like(truth)
if self.frequency_mask_percent_mask > 0: if self.random() > .5:
# Frequency mask # Frequency mask
cs = random.randint(0, c-10) cs = random.randint(0, c-10)
ce = min(c-1, cs+random.randint(1, int(self.frequency_mask_percent_mask*c))) ce = min(c-1, cs+random.randint(1, int(self.frequency_mask_percent_mask*c)))