-
Notifications
You must be signed in to change notification settings - Fork 34
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
getting error on line line 148 in main.py #8
Comments
Hi, thank you for your interest in our work. I will have updated code in a few days time, including a new pre-trained model. |
Yes, I am using the updated commit. Here is the command I use which gives the error |
Issue now resolved, please see the latest code pushed to the repository today. I have also uploaded a model from epoch 99 that is more accurate (better PSNR, SSIM). |
Thank you very much for the fix! By the way, does your code support only GPU, or I can also run on CPU mode? |
Hi, CPU mode is straightforward. You just need to place the cuda keywords in the code with cpu I believe e.g. net.cuda() with net.cpu() etc. For inference you'll also need to load the model on the CPU, more information is here: https://pytorch.org/tutorials/beginner/saving_loading_models.html. Feel free to raise a PR if you have a CPU/GPU switch implemented for CURL. |
Got it, thank you! |
Hi,
Could you please help to run your code without errors.
Right now I have the following issue:
2021-02-23 21:25:34,587 INFO Loading Adobe5k dataset ... Traceback (most recent call last): File "main.py", line 351, in <module> main() File "main.py", line 148, in main inference_dataset = Dataset(data_dict=inference_data_dict, NameError: name 'Dataset' is not defined
The text was updated successfully, but these errors were encountered: