diff --git a/AugmentTests.py b/AugmentTests.py
index e0dc7103870a57aa92613d26c704861062266617..96ffd3678860504db719f85fe4e573cb53f0a860 100644
--- a/AugmentTests.py
+++ b/AugmentTests.py
@@ -16,10 +16,10 @@ if __name__ == "__main__":
         
 
     #Et enfin le rt-shift
+    mean = 0
     records = pd.read_csv("output/perfs.csv")
     for prob in [k/5 for k in range(1,6)]:
         args.augment_args[2] = prob
-        mean = 0
         args.augment_args[5] = mean
         for std in [k/2 for k in range(5,25,5)]:
             args.augment_args[6] = std
diff --git a/config/config.py b/config/config.py
index ae2861d93bbeea900fa59dfbbfb0381c937af0a2..63b5999c22eef360fb137a88c2320ce3282c589d 100644
--- a/config/config.py
+++ b/config/config.py
@@ -5,7 +5,7 @@ def load_args():
     parser = argparse.ArgumentParser()
     parser.add_argument('--epoches', type=int, default=20)
     parser.add_argument('--eval_inter', type=int, default=1)
-    parser.add_argument('--augment_args', nargs = '+', type = float, default = [0,0,0,0.1,0.1,0.,7.5])
+    parser.add_argument('--augment_args', nargs = '+', type = float, default = [0,0,0.5,0.1,0.1,0.,7.5])
     parser.add_argument('--noise_threshold', type=int, default=0)
     parser.add_argument('--lr', type=float, default=0.001)
     parser.add_argument('--optim', type = str, default = "Adam")
diff --git a/output/best_model.pt b/output/best_model.pt
index 89847cc279097a2165b6b9071a44cb8f6aafba07..6a999de51d3f014f67d8325a7cdb023fac52de41 100644
Binary files a/output/best_model.pt and b/output/best_model.pt differ