update

deepke/config.py

@@ -66,7 +66,7 @@ class Config(object):
     batch_size = 64
     train_log = True
     log_interval = 10
-    show_plot = False
+    show_plot = True
     f1_norm = ['macro', 'micro']
 
 

deepke/trainer.py

@@ -36,6 +36,7 @@ def train(epoch, device, dataloader, model, optimizer, criterion, config):
     # plot
     if config.show_plot:
         plt.plot(total_loss)
+        plt.title('loss')
         plt.show()