tnwei / waternet

Modern replication of WaterNet from "An Underwater Image Enhancement Benchmark Dataset and Beyond", IEEE TIP 2019

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

OOM when inference the mp4 file

qimg412 opened this issue · comments

Hello, I am experiencing OOM issues with my RTX a4000 graphics card while using Waternet to reason for MP4 format videos. The video is 10 seconds long and has a resolution of 3840 * 2160. I have multiple RTX a4000 graphics cards, but Waternet does not work for multi card reasoning. Do you have any good solutions?