model changes

piano_transcription/train_model.py

@@ -95,7 +95,7 @@ def run(model, model_name, learn_rate, batch_size, split, k_samples):
         start_time = time.time()
         for batch_nr, f_ins in enumerate(train_batch_iterator):
             cur_loss = train_fn(*f_ins)[0]
-            train_loss_sum += cur_loss[0]
+            train_loss_sum += cur_loss
             train_batches += 1
 
             tr_loss = train_loss_sum / train_batches