projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
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
import torch
from torch import nn
-t = 0
+######################################################################
if len(sys.argv) < 2:
print(sys.argv[0] + ''' <file to monitor>
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)
nn.Conv2d(64, 64, (3, 4))''')
exit(1)
+######################################################################
+
+t = 0
+
while True:
pt = t
t = os.stat(sys.argv[1])[stat.ST_MTIME]
while True:
pt = t
t = os.stat(sys.argv[1])[stat.ST_MTIME]