modify batch_size

Signed-off-by: Your Name <you@example.com>
3 years ago · 12e87d618e
1 changed files with 1 additions and 1 deletions
--- 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