Add 'matplotlib.use('Agg')'; update output path

Train_3Photon.py

@@ -1,5 +1,8 @@
 import sys
 sys.path.append('./src')
+import matplotlib
+matplotlib.use('Agg')
+
 from omegaconf import OmegaConf ### for yaml config parsing
 import torch
 import numpy as np 
@@ -134,6 +137,7 @@ def plot_loss_curves(train_losses, val_losses, test_loss, exp_name, conf):
     plt.grid()
     plotName = f'loss_curve_triplePhoton_{conf.model.version}.png'
     plt.savefig(f'Results/{exp_name}/Plots/{plotName}')
+    plt.close()
 
 def get_model_name(conf):
     modelName = f'triplePhoton{conf.model.version}_{conf.data.energy}keV'