提交 37a4f1f4 编写于 作者: HansBug's avatar HansBug 😆

fix, test(hansbug): fix the bug in torch 1.6.0

上级 dfea6fdd
......@@ -261,7 +261,7 @@ class TestTorchFuncsConstruct:
'x': {
'c': (2, 3, 4),
}
}, 233)
}, 233, dtype=torch.int) # in torch 1.6.0, missing of dtype will raise RuntimeError
assert ttorch.all(_target == 233)
assert _target.shape == ttorch.Size({
'a': torch.Size([2, 3]),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册