提交 42ac7c50 编写于 作者: S shijianning

fix Pylint Warning too-many-function-args

上级 206b36cc
...@@ -50,9 +50,9 @@ if __name__ == "__main__": ...@@ -50,9 +50,9 @@ if __name__ == "__main__":
print("============== Starting Testing ==============") print("============== Starting Testing ==============")
param_dict = load_checkpoint(args.ckpt_path) param_dict = load_checkpoint(args.ckpt_path)
load_param_into_net(network, param_dict) load_param_into_net(network, param_dict)
ds_eval = create_dataset(args.data_path, ds_eval = create_dataset(data_path=args.data_path,
cfg.batch_size, batch_size=cfg.batch_size,
1, repeat_size=1,
"test") status="test")
acc = model.eval(ds_eval, dataset_sink_mode=args.dataset_sink_mode) acc = model.eval(ds_eval, dataset_sink_mode=args.dataset_sink_mode)
print("============== Accuracy:{} ==============".format(acc)) print("============== Accuracy:{} ==============".format(acc))
...@@ -47,10 +47,10 @@ if __name__ == "__main__": ...@@ -47,10 +47,10 @@ if __name__ == "__main__":
model = Model(network, loss, opt, metrics={"Accuracy": Accuracy()}) # test model = Model(network, loss, opt, metrics={"Accuracy": Accuracy()}) # test
print("============== Starting Training ==============") print("============== Starting Training ==============")
ds_train = create_dataset(args.data_path, ds_train = create_dataset(data_path=args.data_path,
cfg.batch_size, batch_size=cfg.batch_size,
cfg.epoch_size, repeat_size=cfg.epoch_size,
"train") status="train")
config_ck = CheckpointConfig(save_checkpoint_steps=cfg.save_checkpoint_steps, config_ck = CheckpointConfig(save_checkpoint_steps=cfg.save_checkpoint_steps,
keep_checkpoint_max=cfg.keep_checkpoint_max) keep_checkpoint_max=cfg.keep_checkpoint_max)
ckpoint_cb = ModelCheckpoint(prefix="checkpoint_alexnet", directory=args.ckpt_path, config=config_ck) ckpoint_cb = ModelCheckpoint(prefix="checkpoint_alexnet", directory=args.ckpt_path, config=config_ck)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册