from local

This commit is contained in:
dohe0342 2022-12-27 11:21:59 +09:00
parent 13ad629c35
commit 529b9f1314
2 changed files with 2 additions and 0 deletions

View File

@ -807,6 +807,8 @@ def compute_loss(
if feature.ndim == 2: if feature.ndim == 2:
feature_lens = [] feature_lens = []
for supervision in supervisions['cut']: for supervision in supervisions['cut']:
print(supervision)
exit()
try: feature_lens.append(supervision.tracks[0].cut.recording.num_samples) try: feature_lens.append(supervision.tracks[0].cut.recording.num_samples)
except: feature_lens.append(supervision.recording.num_samples) except: feature_lens.append(supervision.recording.num_samples)
feature_lens = torch.tensor(feature_lens) feature_lens = torch.tensor(feature_lens)