X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=sizer.py;h=dff36ebc6ba9f279e270339f8a6c548bb8140f86;hp=cc0a19e26d19f5157de277cda7e8b0cea80f1688;hb=eea8998c990d581e94bc6594923efabc37f0f150;hpb=f99e2c83638c960d158c17270c072876834df9a9

diff --git a/sizer.py b/sizer.py
index cc0a19e..dff36eb 100755
--- a/sizer.py
+++ b/sizer.py
@@ -10,7 +10,7 @@ import time
 import torch
 from torch import nn
 
-t = 0
+######################################################################
 
 if len(sys.argv) < 2:
     print(sys.argv[0] + ''' <file to monitor>
@@ -27,6 +27,10 @@ nn.MaxPool2d(5)
 nn.Conv2d(64, 64, (3, 4))''')
     exit(1)
 
+######################################################################
+
+t = 0
+
 while True:
     pt = t
     t = os.stat(sys.argv[1])[stat.ST_MTIME]