sizer.py

   1 #!/usr/bin/env python
   2
   3 import os, stat, sys
   4 import time
   5 import torch
   6 from torch import nn
   7
   8 t = 0
   9
  10 while True:
  11     pt = t
  12     t = os.stat(sys.argv[1])[stat.ST_MTIME]
  13     if t > pt:
  14         pt = t
  15         os.system('clear')
  16         try:
  17             temp = [l.strip('\n\r') for l in open(sys.argv[1], 'r').readlines()]
  18             x = torch.zeros(eval(temp.pop(0)))
  19             print('-> ' + str(tuple(x.size())))
  20             for k in temp:
  21                 print('   ' + k)
  22                 x = eval(k + '(x)')
  23                 print('-> ' + str(tuple(x.size())))
  24         except:
  25             print('** Error **')
  26     time.sleep(1)