Update activation test

examples/alexnet/alexnet.py

@@ -1,6 +1,9 @@
 import torchvision
+import torch
 import sys
 
+from torchsummary import summary
+
 sys.path.append('../../tools')  # Ugly hack 
 from utils import export_model_weights, print_model_parameters
 
@@ -9,5 +12,9 @@ if __name__ == "__main__":
     print_model_parameters(alexnet)  # print layer names and number of parameters
     export_model_weights(alexnet, 'alexnet_weights.bin')
     print()
-    print(alexnet)
+
+    if torch.cuda.is_available():
+        alexnet.cuda()
+
+    summary(alexnet, (3, 227, 227))