lbc12345 / sed Goto Github PK

This error occurred when I was testing, only loading a picture, why

RuntimeError: CUDA out of memory. Tried to allocate 87.95 GiB (GPU 0; 44.48 GiB total capacity; 22.69 GiB already allocated; 19.28 GiB free; 23.92 GiB reserved in total by PyTorch)

Traceback (most recent call last):
File "F:\pythonProject1\SeD-main\train.py", line 15, in
from models import model_rrdb, model_swinir, sed
File "F:\pythonProject1\venv\Lib\site-packages\models_init_.py", line 37, in
import project
ModuleNotFoundError: No module named 'project'
Traceback (most recent call last):
File "F:\pythonProject1\SeD-main\train.py", line 15, in
from models import model_rrdb, model_swinir, sed
File "F:\pythonProject1\venv\Lib\site-packages\models_init_.py", line 37, in
import project
ModuleNotFoundError: No module named 'project'

What is the motivation for adding group normalization in the SeFB module, and what is the purpose of this operation? I did not find any relevant explanations in the article.

Thank you very much for your code sharing, which is very detailed and specific！

How many cards did you use for training and how long did it take.

I seem to need a lot of training time.

Thank you for your contribution. Do you have any testing code?

作者你好，请问代码计划什么时候开源呢？