@@ -0,0 +1,227 @@
+import argparse%0Aimport torchvision.models as models%0A%0A%0Adef get_kernel(params, unparsed_args):%0A    parser = argparse.ArgumentParser(description=%22Benchmark resnet50%22)%0A    parser.parse_args(unparsed_args)%0A    return models.vgg16()%0A

Add vgg16 for pytorch and fix #10