lerobot/examples/real_robot_example/DEBUG_ME.md

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=) 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=) 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')