提交 fcb2b37b 编写于 作者: P pkuliuliu

Fix issue #I1PURI [MA][diff_privacy][Func]the accuracy of diff_privacy...

Fix issue #I1PURI [MA][diff_privacy][Func]the accuracy of diff_privacy tutorial not meet the requirment
上级 84058952
...@@ -22,7 +22,7 @@ mnist_cfg = edict({ ...@@ -22,7 +22,7 @@ mnist_cfg = edict({
'num_classes': 10, # the number of classes of model's output 'num_classes': 10, # the number of classes of model's output
'lr': 0.01, # the learning rate of model's optimizer 'lr': 0.01, # the learning rate of model's optimizer
'momentum': 0.9, # the momentum value of model's optimizer 'momentum': 0.9, # the momentum value of model's optimizer
'epoch_size': 5, # training epochs 'epoch_size': 10, # training epochs
'batch_size': 256, # batch size for training 'batch_size': 256, # batch size for training
'image_height': 32, # the height of training samples 'image_height': 32, # the height of training samples
'image_width': 32, # the width of training samples 'image_width': 32, # the width of training samples
......
...@@ -155,7 +155,7 @@ if __name__ == "__main__": ...@@ -155,7 +155,7 @@ if __name__ == "__main__":
dataset_sink_mode=cfg.dataset_sink_mode) dataset_sink_mode=cfg.dataset_sink_mode)
LOGGER.info(TAG, "============== Starting Testing ==============") LOGGER.info(TAG, "============== Starting Testing ==============")
ckpt_file_name = 'trained_ckpt_file/checkpoint_lenet-5_234.ckpt' ckpt_file_name = 'trained_ckpt_file/checkpoint_lenet-10_234.ckpt'
param_dict = load_checkpoint(ckpt_file_name) param_dict = load_checkpoint(ckpt_file_name)
load_param_into_net(network, param_dict) load_param_into_net(network, param_dict)
ds_eval = generate_mnist_dataset(os.path.join(cfg.data_path, 'test'), ds_eval = generate_mnist_dataset(os.path.join(cfg.data_path, 'test'),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册