search

linjing7 / VR-Baseline

Video Restoration Toolbox including FGST (ICML 2022), S2SVR (ICML 2022), etc.
Apache License 2.0
155 stars 13 forks source link

Train GoPro Datasets error #26

Open MscWood1 opened 1 week ago

MscWood1 commented 1 week ago

Hello, i tried to train gopro dataset with default setting in single GPU. It works well at the first but raised "CUDA out of memory" error when at iter 10000. I wonder how could this happen or how could I modify the setting to avoid this error. Thx for ur answer. error

MscWood1 commented 1 week ago

btw I disabled the validation because it also raises "CUDA out of memory" when it finished a validation and steps into the next training.