Let's use 2 GPUs! /home/csim/anaconda3/envs/pointcept/lib/python3.8/site-packages/torch/optim/lr_scheduler.py:131: UserWarning: Detected call of `lr_scheduler.step()` before `optimizer.step()`. In PyTorch 1.1.0 and later, you should call them in the opposite order: `optimizer.step()` before `lr_scheduler.step()`. Failure to do this will result in PyTorch skipping the first value of the learning rate schedule. See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate warnings.warn("Detected call of `lr_scheduler.step()` before `optimizer.step()`. " Traceback (most recent call last): File "/home/csim/ZSTMSP/MPRNet-main/Deraining/train.py", line 91, in <module> train_dataset = get_training_data(train_dir, {'patch_size':opt.TRAINING.TRAIN_PS}) File "/home/csim/ZSTMSP/MPRNet-main/Deraining/data_RGB.py", line 5, in get_training_data assert os.path.exists(rgb_dir) AssertionError

这段信息中包含了两个部分：

1. 建议使用两个 GPU 进行训练，具体的实现方法可能需要修改代码，具体实现可以参考 PyTorch 的官方文档。

2. 该异常信息与之前的一样，是在 `data_RGB.py` 文件的第 5 行发生的 AssertionError，断言检查 `rgb_dir` 路径是否存在失败。这意味着你在训练模型之前需要先检查输入的文件路径是否正确，并且对应的文件夹是否存在。你可以检查一下相关的代码进行修复。