Update data_factory.py

cure-lab · Mar 15, 2023 · 6fe4c28 · 6fe4c28
1 parent a65bd85
commit 6fe4c28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_provider/data_factory.py b/data_provider/data_factory.py
@@ -17,7 +17,7 @@ def data_provider(args, flag):
 
     if flag == 'test':
         shuffle_flag = False
-        drop_last = True
+        drop_last = False
         batch_size = args.batch_size
         freq = args.freq
     elif flag == 'pred':