diff --git a/output/best_model.pt b/output/best_model.pt
index 7f213f61ad3341d6cef22b361900807adc5700f6..5d79fe1ad168f3fc79ee5d6bcf638c1c59db41a0 100644
Binary files a/output/best_model.pt and b/output/best_model.pt differ
diff --git a/output/perfs.csv b/output/perfs.csv
index 2c74604b3dd3f2fef0074d584e9cb787d19a0979..b43b0e6681014a069c0705b0224815b06ef804e0 100644
--- a/output/perfs.csv
+++ b/output/perfs.csv
@@ -1,52 +1,2 @@
 Augmentation,mu_loss,std_loss,mu_acc,std_acc
-Standard,0.0035194475839777,0.0005053596078284,0.8739837398373986,0.0167605919740555
-rtShift prob0.2 mean0 std2.5,0.0035746556472002,0.0003732012199531,0.8829268292682928,0.0335605974643094
-rtShift prob0.2 mean0 std5.0,0.0037657613676737,0.0004254721619074,0.878048780487805,0.0212006582283012
-rtShift prob0.2 mean0 std7.5,0.0030352800357632,4.6267683792794776e-05,0.8910569105691056,0.0065040650406503
-rtShift prob0.2 mean0 std10.0,0.0037025666333795,0.0008778742655605,0.865040650406504,0.0302456507930703
-rtShift prob0.4 mean0 std2.5,0.0043134410206864,0.0008896470504415,0.853658536585366,0.0391596571668042
-rtShift prob0.4 mean0 std5.0,0.0035770442912249,0.0004097679233287,0.8829268292682926,0.0203089365801573
-rtShift prob0.4 mean0 std7.5,0.0038386337156218,0.0007418083569182,0.8796747967479674,0.0214486275744274
-rtShift prob0.4 mean0 std10.0,0.0036847093725592,0.0005269658048506,0.8747967479674796,0.0209497540271953
-rtShift prob0.6 mean0 std2.5,0.0033642452422196,0.0007419803049781,0.8878048780487804,0.0350254621602244
-rtShift prob0.6 mean0 std5.0,0.0038544148449006,0.0008509633083652,0.865040650406504,0.0274984301224191
-rtShift prob0.6 mean0 std7.5,0.0033022010714057,0.0002617130408073,0.8975609756097562,0.0132098185441235
-rtShift prob0.6 mean0 std10.0,0.0039205908775329,0.0004898444551765,0.8617886178861788,0.0072717657804871
-rtShift prob0.8 mean0 std2.5,0.0036633977560493,0.0003803085043467,0.88130081300813,0.0132098185441235
-rtShift prob0.8 mean0 std5.0,0.0044963194102775,0.0011998353116845,0.8634146341463416,0.0248732659199648
-rtShift prob0.8 mean0 std7.5,0.0037264997881602,0.0004121150381343,0.865040650406504,0.0279750415188378
-rtShift prob0.8 mean0 std10.0,0.004013952346352,0.0005384237498584,0.8666666666666666,0.0221759051983509
-rtShift prob1.0 mean0 std2.5,0.0042933197525458,0.000356641602263,0.88130081300813,0.0209497540271953
-rtShift prob1.0 mean0 std5.0,0.0039978255101335,0.0007889839396288,0.8715447154471546,0.0165821772799765
-rtShift prob1.0 mean0 std7.5,0.0043511592760318,0.0008649844557656,0.8764227642276422,0.0188224990289271
-rtShift prob1.0 mean0 std10.0,0.0038671349122272,0.0007948168302068,0.8878048780487806,0.0248732659199648
-intShift prob0.2 max1.25,0.004374882748456504,0.0006700828902289184,0.8699186991869918,0.011497671238805665
-intShift prob0.2 max1.5,0.0036952993734096122,0.0004379449804924912,0.8764227642276424,0.025914434878876794
-intShift prob0.2 max1.75,0.004274448388960303,0.0005615522502595648,0.878048780487805,0.022413087401772693
-intShift prob0.2 max2.0,0.0038483763128761353,0.00033108248369751234,0.8764227642276422,0.016582177279976528
-intShift prob0.2 max2.25,0.00429740179844988,0.0008211964604526267,0.8845528455284553,0.016582177279976545
-intShift prob0.2 max2.5,0.0037177296188788686,0.0004946500202442801,0.8829268292682928,0.016740862017865055
-intShift prob0.4 max1.25,0.0040755013624827075,0.0013897315577589401,0.8943089430894309,0.021200658228301302
-intShift prob0.4 max1.5,0.0037607090259955183,0.000621610225749857,0.8796747967479674,0.01658217727997656
-intShift prob0.4 max1.75,0.004143828638200837,0.0009279966167386837,0.8861788617886178,0.011497671238805665
-intShift prob0.4 max2.0,0.004241930760019193,0.00037692032654743806,0.8829268292682926,0.026519522650894453
-intShift prob0.4 max2.25,0.003895949445119718,0.0002878210124066211,0.8569105691056912,0.014175281117205457
-intShift prob0.4 max2.5,0.004080777924235274,0.0006749719203261898,0.8845528455284553,0.020178331131692446
-intShift prob0.6 max1.25,0.003982379814473594,0.0009096948428901936,0.873170731707317,0.015079054464220663
-intShift prob0.6 max1.5,0.004085246818821605,0.0005302856829458952,0.878048780487805,0.02241308740177271
-intShift prob0.6 max1.75,0.004415224044303583,0.0003882054923378211,0.8747967479674796,0.009756097560975634
-intShift prob0.6 max2.0,0.004203973262290645,0.00066407077666088,0.8780487804878048,0.025709574472913638
-intShift prob0.6 max2.25,0.00380666452210124,0.0009967767081236647,0.8959349593495934,0.027402113083500355
-intShift prob0.6 max2.5,0.0037946056059705536,0.0003843266282190126,0.8650406504065039,0.019647229225357023
-intShift prob0.8 max1.25,0.0038289718996218553,0.0006457646696712747,0.8829268292682926,0.011028178834350023
-intShift prob0.8 max1.5,0.004250375042116739,0.0008156539590035773,0.8715447154471544,0.02835056223441089
-intShift prob0.8 max1.75,0.004625961838698968,0.0008039631675386016,0.8487804878048781,0.03110752271497396
-intShift prob0.8 max2.0,0.0033499934808994694,0.0006807762714168269,0.8959349593495937,0.025914434878876808
-intShift prob0.8 max2.25,0.0038570141889215487,0.0009787725490736357,0.8991869918699187,0.014175281117205438
-intShift prob0.8 max2.5,0.003945898282818678,0.0006875475195834033,0.8975609756097562,0.015931640603467842
-intShift prob1.0 max1.25,0.003927457284152023,0.0005794441787879437,0.8861788617886178,0.005141914894582723
-intShift prob1.0 max1.5,0.004095270720923819,0.0008008379709253904,0.8764227642276422,0.016582177279976528
-intShift prob1.0 max1.75,0.003849132468060749,0.0004707415154935753,0.861788617886179,0.02908706312194849
-intShift prob1.0 max2.0,0.004305678606033325,0.0004519948084980065,0.8699186991869918,0.013604228073724826
-intShift prob1.0 max2.25,0.004030883990652193,0.0008368862937073165,0.8926829268292684,0.02881308153929976
-intShift prob1.0 max2.5,0.003993577055814789,0.0006535206765319719,0.8747967479674796,0.018251987252555814
+Standard,0.0055081038455653,0.0010060043637088575,0.8191056910569106,0.04064024263179612