Wondering which graphics did you experiment on?

ZJZAC / Deep-Model-Watermarking

MIT License

68 stars 9 forks source link

Wondering which graphics did you experiment on? #2

Closed RorschachChen closed 2 years ago

RorschachChen commented 2 years ago

Hi Dear ZJ, Great Work! I am trying to run with single RTX 2080Ti 11G, but the BatchSize I need to set as 2 to run, otherwise OOM problem is triggered? I noticed the BatchSize displayed in the code is 16.

ZJZAC commented 2 years ago

Hi Chen，Thanks for your interest to our work. We used 4 1080Ti/2080Ti in a parallel way to solve the OOM problem with BatchSize 16. BTW, the size of input image or the size of models can also be shrinked to overcome OOM.