Antoine Louis

Results 4 comments of Antoine Louis

Hi @jeffra, yes I'm experiencing the same issue. Here is the error I get: ``` File "/root/envs/star/lib/python3.8/site-packages/grad_cache/grad_cache.py", line 242, in forward_backward surrogate.backward() File "/root/envs/star/lib/python3.8/site-packages/torch/_tensor.py", line 307, in backward torch.autograd.backward(self, gradient,...

Hi @EParisot, did you find a solution to this issue?

Amazing, thanks for sharing your solution @EParisot!