try fix pylint warning

pull/995/head
shijianning 5 years ago
parent 710a0e817b
commit 5afcefdce4

@ -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(data_path=args.data_path, ds_eval = create_dataset(args.data_path,
batch_size=cfg.batch_size, cfg.batch_size,
repeat_size=1, 1,
status="test") "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__":
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(data_path=args.data_path, ds_train = create_dataset(args.data_path,
batch_size=cfg.batch_size, cfg.batch_size,
repeat_size=cfg.epoch_size, cfg.epoch_size,
status="train") "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)

Loading…
Cancel
Save