diff --git a/config/config.py b/config/config.py
index 406b5af4bad923b2286e7ca4dfe6226ce4d4e33b..789f0bc1a495a8581be2fb0087b012598b3d7233 100644
--- a/config/config.py
+++ b/config/config.py
@@ -4,7 +4,7 @@ import argparse
 def load_args():
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('--epoches', type=int, default=100)
+    parser.add_argument('--epoches', type=int, default=10)
     parser.add_argument('--save_inter', type=int, default=50)
     parser.add_argument('--eval_inter', type=int, default=1)
     parser.add_argument('--noise_threshold', type=int, default=0)
diff --git a/main.py b/main.py
index ff27a40e75a7eb057e321c27cf81f770fff2ac10..6ef07536e7096d4f2f3bacb3c49e23114cd27000 100644
--- a/main.py
+++ b/main.py
@@ -235,6 +235,7 @@ def make_prediction_duo(model, data, f_name):
     classes = data.dataset.dataset.classes
     print('Prediction made')
     # Build confusion matrix
+    print(len(y_true),len(y_pred))
     cf_matrix = confusion_matrix(y_true, y_pred)
     print('CM made')
     df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],