Commit 5b78432f authored by sjjsmuel's avatar sjjsmuel

rename

parent 7a5f3b8c
......@@ -52,9 +52,9 @@ time = get_curr_time()
checkpoint_path = pathlib.Path(options.output_path +'/checkpoints/{}'.format(time))
if not checkpoint_path.exists():
checkpoint_path.mkdir()
resnet_file = None
base_model_file = None
if options.base_net_file:
resnet_file = options.base_net_file
base_model_file = options.base_net_file
img_width = options.width
img_height = options.height
channels = 3
......@@ -99,7 +99,7 @@ test_dataset = test_loader.load_dataset()
# Create Network
network = VGG_16(n_classes, img_width, img_height, channels)
network = VGG_16(n_classes, img_width, img_height, channels, base_model_file)
model = network.get_model()
#compile the model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment