update examples

ricky40403 · Oct 19, 2019 · a703072 · a703072
1 parent c466ef3
commit a703072
Show file tree

Hide file tree

Showing 16 changed files with 1,874 additions and 2,904 deletions.
diff --git a/examples/alexnet.png b/examples/alexnet.png
diff --git a/examples/densenet161.png b/examples/densenet161.png
diff --git a/examples/example.ipynb b/examples/example.ipynb
diff --git a/examples/googlenet.png b/examples/googlenet.png
diff --git a/examples/inception_v3.png b/examples/inception_v3.png
diff --git a/examples/mobilenet_v2.png b/examples/mobilenet_v2.png
diff --git a/examples/resnet18.png b/examples/resnet18.png
diff --git a/examples/shufflenet_v2_x1_0.png b/examples/shufflenet_v2_x1_0.png
diff --git a/examples/squeezenet1_0.png b/examples/squeezenet1_0.png
diff --git a/examples/vgg16.png b/examples/vgg16.png
diff --git a/summary_example.ipynb b/summary_example.ipynb
diff --git a/test.py b/test.py
@@ -4,13 +4,19 @@
 import torchvision.models as models
 import copy
 from transformers.torchTransformer import TorchTransformer
-# from quantize import QConv2d
-model = models.__dict__["inception_v3"]()
-# model.cuda()
+from transformers.quantize import QConv2d
+model = models.__dict__["resnet18"]()
+model.cuda()
 model = model.eval()
 
 transofrmer = TorchTransformer()
+transofrmer.register(nn.Conv2d, QConv2d)
+model = transofrmer.trans_layers(model)
+print(model)
+sys.exit()
+
+
 input_tensor = torch.randn([1, 3, 224, 224])		
-# input_tensor = input_tensor.cuda()		
+input_tensor = input_tensor.cuda()
 net = transofrmer.summary(model, input_tensor=input_tensor)
-# transofrmer.visualize(model, save_name= "example", graph_size = 80)
+# transofrmer.visualize(model, input_tensor = input_tensor, save_name= "example", graph_size = 80)
diff --git a/transform_example.ipynb b/transform_example.ipynb