From 12e87d618ef31fc13e404d3a656d3d7afa87906d Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 23 Feb 2022 17:29:11 +0800 Subject: [PATCH] modify batch_size Signed-off-by: Your Name --- resnet_training_yaml.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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