revlib

revlib copied to clipboard

- [ ] wait for Torch 1.11 to be released - [x] Avoid Torch 1.10 crashing - [ ] Extend ReadMe - [ ] give credit

ClashLuke

revlib can not work in torch amp.

8

dear authors, when using revlib in torch amp, it reports error as follow: Traceback (most recent call last): File "/home/fqm/.conda/envs/torch/lib/python3.9/site-packages/revlib/core.py", line 130, in backward mod_out = take_0th_tensor(new_mod.wrapped_module(y0, *ctx.args, **ctx.kwargs)) File...

JAYatBUAA