Rishik Mourya
Rishik Mourya
Declaring the model to the GPU solves the issue, ```python class Classifier(nn.Module): def __init__(self): super(Classifier, self).__init__() self.model = nn.Sequential( nn.Conv2d(1, 8, 3, 1), # (-1, 8, 26, 26) nn.LeakyReLU(0.2), nn.MaxPool2d(2),...
> both model and input ported on gpu using model.to(device) and image.to(device). Availability of gpu is also verified. Still getting the error. Kindly help if any solution Try passing `device`...
Yes, the code is working as expected, it's just that the processing would be slow since no GPU support is there in your system.
Added
Same error here!
Closing since this error is not part of the project, prefer to your system based python installation guides and forums.
Can you send some screenshots? Quality reduction should be unnoticeable for the regions where there is no watermark.
@friki67 that's the correct fix, could you make the pull request?
I have to look into it, I think model is not performing well for all black input... I gotta test a bit and look into its potential solution. Thanks for...