From 1e0d48d059ab8c38f0d65cce169a57af2466d63b Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 28 Nov 2022 13:23:19 +0100 Subject: [PATCH] Update. --- stack.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/stack.py b/stack.py index 27de204..453e9ed 100755 --- a/stack.py +++ b/stack.py @@ -11,13 +11,13 @@ import sys def exception_hook(exc_type, exc_value, tb): - tb = tb.tb_next +# tb = tb.tb_next while tb: - #x=tb.tb_frame.f_code + # x=tb.tb_frame #.f_code # for field in dir(x): # print(f'@@@ {field} {getattr(x, field)}') - + print('--------------------------------------------------') filename = tb.tb_frame.f_code.co_filename name = tb.tb_frame.f_code.co_name line_no = tb.tb_lineno @@ -28,7 +28,7 @@ def exception_hook(exc_type, exc_value, tb): for n,v in local_vars.items(): if is_tensor(v): - print(f' {n} -> {v.size()}:{v.dtype}:{v.device}') + print(f' {n} -> {tuple(v.size())}:{v.dtype}:{v.device}') else: print(f' {n} -> {v}') @@ -51,8 +51,9 @@ if __name__ == '__main__': c=b+b dummy(a,c) - m=torch.randn(2,3) - x=torch.randn(3) - blah(m,x) - blah(x,m) + mmm=torch.randn(2,3) + xxx=torch.randn(3) + #print(xxx@mmm) + blah(mmm,xxx) + blah(xxx,mmm) -- 2.39.5