X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=sizer.py;fp=sizer.py;h=52620e88aac6600f5d5e80400a7162abc84991b2;hb=8a6b6efe651113cc3e8eb13cb13059724948bc9d;hp=0000000000000000000000000000000000000000;hpb=d29423af71d76b52cbdf04367dbfe1908a492786;p=pytorch.git

diff --git a/sizer.py b/sizer.py
new file mode 100755
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)