forked from OSSInnovation/mindspore
remove unused param `buffer_size`
This commit is contained in:
parent
57ecb40022
commit
5b4f6621ec
|
@ -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
|
||||
|
|
|
@ -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/',
|
||||
|
|
Loading…
Reference in New Issue