Browse Source
Debug decorator
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
2 changed files with
3 additions and
1 deletions
-
test_onnx.py
-
timm_image.py
|
|
@ -57,6 +57,8 @@ for name in models: |
|
|
status = [name] + ['fail'] * 5 |
|
|
status = [name] + ['fail'] * 5 |
|
|
|
|
|
|
|
|
try: |
|
|
try: |
|
|
|
|
|
out0 = op(data.cpu().detach().numpy().squeeze(0)) |
|
|
|
|
|
assert out0.shape |
|
|
out1 = op.model(data).detach().numpy() |
|
|
out1 = op.model(data).detach().numpy() |
|
|
logger.info('OP LOADED.') |
|
|
logger.info('OP LOADED.') |
|
|
status[1] = 'success' |
|
|
status[1] = 'success' |
|
|
|
|
|
@ -44,7 +44,7 @@ log = logging.getLogger('timm_op') |
|
|
def torch_no_grad(f): |
|
|
def torch_no_grad(f): |
|
|
def wrap(*args, **kwargs): |
|
|
def wrap(*args, **kwargs): |
|
|
with torch.no_grad(): |
|
|
with torch.no_grad(): |
|
|
f(*args, **kwargs) |
|
|
|
|
|
|
|
|
return f(*args, **kwargs) |
|
|
return wrap |
|
|
return wrap |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
