diff --git a/model_zoo/official/cv/googlenet/README.md b/model_zoo/official/cv/googlenet/README.md index 013ae86b45..a7e5e21851 100644 --- a/model_zoo/official/cv/googlenet/README.md +++ b/model_zoo/official/cv/googlenet/README.md @@ -163,7 +163,6 @@ Parameters for both training and evaluation can be set in config.py 'epoch_size': 125 # total training epochs 'momentum': 0.9 # momentum 'weight_decay': 5e-4 # weight decay value - 'buffer_size': 10 # buffer size 'image_height': 224 # image height used as input to the model 'image_width': 224 # image width used as input to the model 'data_path': './cifar10' # absolute full path to the train and evaluation datasets diff --git a/model_zoo/official/cv/googlenet/src/config.py b/model_zoo/official/cv/googlenet/src/config.py index 2989e2ba69..a080969299 100644 --- a/model_zoo/official/cv/googlenet/src/config.py +++ b/model_zoo/official/cv/googlenet/src/config.py @@ -26,7 +26,6 @@ cifar_cfg = edict({ 'epoch_size': 125, 'momentum': 0.9, 'weight_decay': 5e-4, - 'buffer_size': 10, 'image_height': 224, 'image_width': 224, 'data_path': './cifar10', @@ -47,7 +46,6 @@ imagenet_cfg = edict({ 'epoch_size': 300, 'momentum': 0.9, 'weight_decay': 1e-4, - 'buffer_size': None, # invalid parameter 'image_height': 224, 'image_width': 224, 'data_path': './ImageNet_Original/train/',