Skip to content
Snippets Groups Projects
Commit 4014c9f5 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Bump predtuner version (bugfix) and changes to test script

parent 65bb1fea
No related branches found
No related tags found
No related merge requests found
Subproject commit a149e365170263666db764664ad8ed6b03f258d3 Subproject commit d65bd5c8b774a3b90e983aba7763342e38757421
...@@ -13,8 +13,8 @@ from torch.nn import Module ...@@ -13,8 +13,8 @@ from torch.nn import Module
site.addsitedir(Path(__file__).parent.parent.absolute()) site.addsitedir(Path(__file__).parent.parent.absolute())
import dnn import dnn
# Set up logger to put log file in /tmp # Set up logger
msg_logger = config_pylogger(output_dir="/tmp", verbose=True) msg_logger = config_pylogger(output_dir=".", verbose=True)
benchmarks = { benchmarks = {
"lenet_mnist": (dnn.LeNet, 1, 28, 1000), "lenet_mnist": (dnn.LeNet, 1, 28, 1000),
"alexnet_imagenet": (dnn.AlexNetImageNet, 3, 224, 100), "alexnet_imagenet": (dnn.AlexNetImageNet, 3, 224, 100),
...@@ -59,7 +59,6 @@ def main(): ...@@ -59,7 +59,6 @@ def main():
binary_file, metadata_file = generate( binary_file, metadata_file = generate(
model_cls, nch, img_size, batch_size, netname model_cls, nch, img_size, batch_size, netname
) )
print(Path.cwd(), binary_file, metadata_file)
app = PipedBinaryApp("test", binary_file, metadata_file) app = PipedBinaryApp("test", binary_file, metadata_file)
tuner = app.get_tuner() tuner = app.get_tuner()
tuner.tune( tuner.tune(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment