Update.

[pytorch.git] / sizer.py

diff --git a/sizer.py b/sizer.py
index cc0a19e..dff36eb 100755 (executable)
--- 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]