|
|
@ -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)
|
|
|
|