Update modeling_super_linear.py

modeling_super_linear.py

@@ -261,6 +261,7 @@ class RLinear(nn.Module):
 "-------------------------------------------------------------------------------------------------------------------"
 class SparseNoisyMoE(nn.Module):
     def __init__(self, configs, experts=None):
+        self.i = 0
         super(SparseNoisyMoE, self).__init__()
         input_dim = configs.seq_len
         self.lookback = configs.seq_len
@@ -429,7 +430,8 @@ class SparseNoisyMoE(nn.Module):
                  verticalalignment='top', bbox=props)
 
         plt.tight_layout()
-        plt.savefig('expert_probabilities.png', bbox_inches='tight')
+        plt.savefig(F"expert_probabilities_{self.i}.png", bbox_inches='tight')
+        self.i+=1
         plt.close()
         print(expert_probs.shape)