Update.

[pytorch.git] / sizer.py

diff --git a/sizer.py b/sizer.py
new file mode 100755 (executable)
index 0000000..52620e8
--- /dev/null
+++ b/sizer.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+import os, stat, sys
+import time
+import torch
+from torch import nn
+
+t = 0
+
+while True:
+    pt = t
+    t = os.stat(sys.argv[1])[stat.ST_MTIME]
+    if t > pt:
+        pt = t
+        os.system('clear')
+        try:
+            temp = [l.strip('\n\r') for l in open(sys.argv[1], 'r').readlines()]
+            x = torch.zeros(eval(temp.pop(0)))
+            print('-> ' + str(tuple(x.size())))
+            for k in temp:
+                print('   ' + k)
+                x = eval(k + '(x)')
+                print('-> ' + str(tuple(x.size())))
+        except:
+            print('** Error **')
+    time.sleep(1)