-
Notifications
You must be signed in to change notification settings - Fork 30
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
task_name #40
Comments
Hello! Where is this code? (I forget why but I guess I wanted to print whether the predicted degradation type is correct.) |
import torch checkpoint = 'pretrained/daclip_ViT-B-32.pt' image = preprocess(Image.open("haze_01.png")).unsqueeze(0) with torch.no_grad(), torch.cuda.amp.autocast():
print(f"Task: {task_name}: {degradations[index]} - {text_probs[0][index]}")there |
Ok, according to the code, I think I just want to know whether the predicted degradation type is correct. The task name should be pre-defined like |
Hello! What does it mean to achieve task_name in print(f"Task: {task_name}: {index]} - {text_probs[0][index]}")?
The text was updated successfully, but these errors were encountered: