diff --git a/resnet_training_yaml.yaml b/resnet_training_yaml.yaml
index 1a79cfe..b02ff26 100644
--- a/resnet_training_yaml.yaml
+++ b/resnet_training_yaml.yaml
@@ -26,7 +26,7 @@ logging:
 metrics:
   metric: Accuracy
 train:
-  batch_size: 256
+  batch_size: 16
   dataloader_drop_last: false
   dataloader_num_workers: 0
   dataloader_pin_memory: true