Antoine Dumoulin
Antoine Dumoulin
callback would not take into arguments if kwargs is present
```python def foo(a:torch.Tensor, b:torch.Tensor): bool_mat = (a & b) print(bool_mat) return bool_mat.nonzero_static(size=bool_mat.shape[0]) a = torch.zeros((5,5), dtype=torch.bool, device="cuda") b = a.clone() res = torch.func.vmap(foo)(a, b) ``` ErrorCode: ``` NotImplementedError: Could not...
I have fit SMPL female parameters to the `neutral_body/mean_all.obj` of GarmentCodeData. I'm not sure if I can share them here for licensing reason but I think it could be useful...