-
Notifications
You must be signed in to change notification settings - Fork 21
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to load pretrained Enhancement model in maxim_torch.py #9
Comments
I met the same question. Have you addressed it? |
No, this repo is dead. We reverse engineered the code and implemented it ourselves in pytorch. Got really bad results on large resolutions and wont use this in the future |
I downloaded weights from maxim repo
Then in maxim-pytorch repo i run jax2torch.py file with
python maxim_pytorch/jax2torch.py -c maxim_ckpt_Enhancement_FiveK_checkpoint.npz
It works and I get torch_weight.pth file
I then try to load it but im unable to understand if Im giving the wrong arguments or your code is wrong
I get error:
The text was updated successfully, but these errors were encountered: