forked from tangger/lerobot
15 lines
956 B
Markdown
15 lines
956 B
Markdown
inp = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_inp.pt')
|
|
conv = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_conv.pt')
|
|
out = torch.nn.functional.conv2d(inp, conv, bias=None, stride=1, padding=1, dilation=1, groups=1)
|
|
d = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_out.pt')
|
|
print((out-d).abs().max())
|
|
tensor(0.0044, device='cuda:0', grad_fn=<MaxBackward1>)
|
|
inp = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_inp.pt').to('cpu')
|
|
conv = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_conv.pt').to('cpu')
|
|
out = torch.nn.functional.conv2d(inp, conv, bias=None, stride=1, padding=1, dilation=1, groups=1)
|
|
d = torch.load('/home/thomwolf/Documents/Github/ACT/tensor_out.pt')
|
|
print((out-d).abs().max())
|
|
tensor(0., grad_fn=<MaxBackward1>)
|
|
out = torch.nn.functional.conv2d(inp, conv, bias=None, stride=1, padding=1, dilation=1, groups=1)
|
|
torch.save(out, '/home/thomwolf/Documents/Github/ACT/tensor_out_lerobot.pt')
|