va1shn9v / PromptIR

PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
https://arxiv.org/abs/2306.13090
Other
328 stars 24 forks source link

Hello, senior! I have some questions about PromptIR that I don't understand. Could you please help me with them? #23

Closed Threegood-student closed 5 months ago

Threegood-student commented 9 months ago

Hello, I apologize for taking up your valuable time. I have learned a lot from your article "PromptIR: Prompting for All-in-One Blind Image Restoration." However, I encountered some issues when trying to reproduce the results, and I hope to seek your assistance.

After running the command "conda env create -f env.yml," I encountered a "ModuleNotFoundError: No module named 'lightning'" when running train.py. Subsequently, I installed it using pip install lightning==2.0.1. However, a new issue arose with "AttributeError: module 'torch' has no attribute 'autocast'." I am unsure how to resolve this.

I would greatly appreciate your help!