fix https://github.com/RVC-Boss/GPT-SoVITS/issues/2250

fix https://github.com/RVC-Boss/GPT-SoVITS/issues/2250

GPT_SoVITS/s2_train.py

@@ -429,6 +429,8 @@ def train_and_evaluate(
                 # scalar_dict.update({"loss/g/{}".format(i): v for i, v in enumerate(losses_gen)})
                 # scalar_dict.update({"loss/d_r/{}".format(i): v for i, v in enumerate(losses_disc_r)})
                 # scalar_dict.update({"loss/d_g/{}".format(i): v for i, v in enumerate(losses_disc_g)})
+                image_dict=None
+                try:###Some people installed the wrong version of matplotlib.
                     image_dict = {
                         "slice/mel_org": utils.plot_spectrogram_to_numpy(
                             y_mel[0].data.cpu().numpy()
@@ -443,12 +445,9 @@ def train_and_evaluate(
                             stats_ssl[0].data.cpu().numpy()
                         ),
                     }
-                utils.summarize(
+                except:pass
-                    writer=writer,
+                if image_dict:utils.summarize(writer=writer,global_step=global_step,images=image_dict,scalars=scalar_dict,)
-                    global_step=global_step,
+                else:utils.summarize(writer=writer,global_step=global_step,scalars=scalar_dict,)
-                    images=image_dict,
-                    scalars=scalar_dict,
-                )
         global_step += 1
     if epoch % hps.train.save_every_epoch == 0 and rank == 0:
         if hps.train.if_save_latest == 0: