feat: increased grain worker buffer size

Author: AshishKumar4

training.py

@@ -92,7 +92,7 @@ def boolean_string(s):
 parser.add_argument('--GRAIN_READ_BUFFER_SIZE', type=int,
                     default=96, help='Grain read buffer size')
 parser.add_argument('--GRAIN_WORKER_BUFFER_SIZE', type=int,
-                    default=40, help='Grain worker buffer size')
+                    default=100, help='Grain worker buffer size')
 
 parser.add_argument('--batch_size', type=int, default=32, help='Batch size')
 parser.add_argument('--image_size', type=int, default=128, help='Image size')