Your project is very enlightening, but I have a doubt, this code runs in the configuration file environment.yaml you provided, the project does not run successfully, it will show an error like this:
Traceback (most recent call last):
File "D:\17DEM\LDM-SRtuning-main\main.py", line 6, in
import pytorch_lightning as pl
File "D:\Anaconda\envs\diffusion\lib\site-packages\pytorch_lightning_init.py", line 20, in
from pytorch_lightning import metrics # noqa: E402
File "D:\Anaconda\envs\diffusion\lib\site-packages\pytorch_lightning\metrics_init_.py", line 15, in
from pytorch_lightning.metrics.classification import ( # noqa: F401
File "D:\Anaconda\envs\diffusion\lib\site-packages\pytorch_lightning\metrics\classification_init_.py", line 14, in
from pytorch_lightning.metrics.classification.accuracy import Accuracy # noqa: F401
File "D:\Anaconda\envs\diffusion\lib\site-packages\pytorch_lightning\metrics\classification\accuracy.py", line 16, in
from torchmetrics import Accuracy as Accuracy
File "D:\Anaconda\envs\diffusion\lib\site-packages\torchmetrics_init.py", line 14, in
from torchmetrics import functional # noqa: E402
File "D:\Anaconda\envs\diffusion\lib\site-packages\torchmetrics\functional_init_.py", line 14, in
from torchmetrics.functional.audio.deprecated import permutation_invariant_training as permutation_invariant_training
File "D:\Anaconda\envs\diffusion\lib\site-packages\torchmetrics\functional\audio_init.py", line 15, in
from torchmetrics.functional.audio.sdr import (
File "D:\Anaconda\envs\diffusion\lib\site-packages\torchmetrics\functional\audio\sdr.py", line 27, in
solve = torch.linalg.solve
AttributeError: module 'torch.linalg' has no attribute 'solve'
I solved it, but I can't call GPU training:
GPU available: True, used: False
TPU available: False, using: 0 TPU cores
IPU available: False, using: 0 IPUs