Update vivit.py

TomStog · Mar 3, 2023 · acde2fe · acde2fe
1 parent cfde829
commit acde2fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vivit.py b/vivit.py
@@ -253,7 +253,7 @@ def run_experiment():
     mae_total = []
     mse_total = []
 
-    for i in range(5):
+    for i in range(20):
         mae = []
         mae_temp = []
         mse = []