Pretrained Model

[joaocps]

Congratulations for the excellent work!

Because I don't have computational capacity, could you provide the pre-trained model in order to do some tests?

[joaocps]

@MRUIL

[MRUIL]

Congratulations for the excellent work!

Because I don't have computational capacity, could you provide the pre-trained model in order to do some tests?

Hi @joaocps , for some reason, I didn’t save the pre-trained model, I really apologize.

[joaocps]

Hi again @MRUIL

When i run the test.py with the pre-trained model and one watermarked image raised an error like:

RuntimeError: CUDA out of memory. Tried to allocate 254.00 MiB (GPU 0; 4.00 GiB total capacity; 2.95 GiB already allocated; 0 bytes free; 2.98 GiB reserved in total by PyTorch)

Any possible solution?

[joaocps]

@MRUIL

[joaocps]

@MRUIL

[MRUIL]

@MRUIL

Hi @joaocps , your GPU memory is too small, you need to change another device. Or you could try to reduce the input resolution to 200*200, but I'm not sure whether it's OK.