ValueError: Expected more than 1 spatial element when training, got input size torch.Size([1, 512, 1, 1])

[xcfstudio]

raise ValueError("Expected more than 1 spatial element when training, got input size {}".format(size))
ValueError: Expected more than 1 spatial element when training, got input size torch.Size([1, 512, 1, 1])

analytics-python 1.4.0
backoff 1.10.0
bcrypt 3.2.0
certifi 2021.10.8
cffi 1.15.0
charset-normalizer 2.0.8
click 8.0.3
colorama 0.4.4
cryptography 36.0.0
cycler 0.11.0
ffmpy 0.3.0
Flask 2.0.2
Flask-CacheBuster 1.0.0
Flask-Cors 3.0.10
Flask-Login 0.5.0
fonttools 4.28.2
gradio 2.4.6
idna 3.3
itsdangerous 2.0.1
Jinja2 3.0.3
kiwisolver 1.3.2
markdown2 2.4.1
MarkupSafe 2.0.1
matplotlib 3.5.0
monotonic 1.6
numpy 1.21.4
packaging 21.3
pandas 1.3.4
paramiko 2.8.1
pep517 0.12.0
Pillow 8.4.0
pip 21.3.1
pycparser 2.21
pycryptodome 3.11.0
pydub 0.25.1
PyNaCl 1.4.0
pyparsing 3.0.6
python-dateutil 2.8.2
pytz 2021.3
requests 2.26.0
setuptools 41.2.0
setuptools-scm 6.3.2
six 1.16.0
tomli 1.2.2
torch 1.10.0
torchtext 0.11.0
torchvision 0.11.1
tqdm 4.62.3
typing_extensions 4.0.0
urllib3 1.26.7
Werkzeug 2.0.2