From 4d1d04a6a03a270e867c12ece86418988867bee9 Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Thu, 11 Jun 2020 10:48:20 +0200
Subject: [PATCH] bug fix

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index f6d73a4a..8581132a 100644
--- a/train.py
+++ b/train.py
@@ -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()