projects
/
pytorch.git
/ blob
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
history
|
raw
|
HEAD
Update.
[pytorch.git]
/
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)