bug fix

2026-09-09 11:16:00 +08:00 · 2020-06-15 19:04:05 +02:00
parent e8a59db839
commit b8766c05ce
1 changed files with 1 additions and 1 deletions
@@ -237,7 +237,7 @@ def train(model, criterion, optimizer, optimizer_st, scheduler,
                    # save model
                    save_checkpoint(model, optimizer, global_step, epoch, model.decoder.r, OUT_PATH,
                                    optimizer_st=optimizer_st,
-                                    model_loss=loss_dict['postnet_loss'].item())
+                                    model_loss=loss_dict['postnet_loss'])

                # Diagnostic visualizations
                const_spec = postnet_output[0].data.cpu().numpy()