-
Notifications
You must be signed in to change notification settings - Fork 484
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
wrap image write in try #175
Conversation
d199b73
to
2737517
Compare
Are CI failures related to the pytorch version? What can we do to get deepvoice working on the latest? |
1 similar comment
Are CI failures related to the pytorch version? What can we do to get deepvoice working on the latest? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sorry for the delay... Looks better now. Thank you! I will look into the CI failures later. I will merge the PR once my last comments are resolved.
train.py
Outdated
mel_output = prepare_spec_image(audio._denormalize(mel_output)) | ||
writer.add_image("Predicted mel spectrogram", mel_output, global_step) | ||
try: | ||
mel_output = mel_outputs[idx].cpu().data.numpy() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you move this line to the out of try block? Seems irrelevant.
train.py
Outdated
# Save averaged alignment | ||
alignment_dir = join(checkpoint_dir, "alignment_ave") | ||
os.makedirs(alignment_dir, exist_ok=True) | ||
path = join(alignment_dir, "step{:09d}_layer_{}_alignment.png".format( | ||
global_step, i + 1)) | ||
path = join(alignment_dir, | ||
"step{:09d}_alignment.png".format(global_step)) | ||
alignment = attn.mean(0)[idx].cpu().data.numpy() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please move the code to out of try block.
This sometimes fails due to "Unsupported format" error. Report the exception and move on.
2737517
to
68296f7
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you so much for your contribution!!
This sometimes fails due to "Unsupported format" error.
Report the exception and move on.