pretrained model RuntimeError

[miraclerepuduce]

Very stunning work. But I found a problem when I used the checkpoint you provided for testing. RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase.

    This probably means that you are not using fork to start your
    child processes and you have forgotten to use the proper idiom
    in the main module:

        if __name__ == '__main__':
            freeze_support()
            ...

    The "freeze_support()" line can be omitted if the program
    is not going to be frozen to produce an executable.

Traceback (most recent call last): File "E:\softwareyang\envs\pytorch_env3.10\lib\site-packages\torch\utils\data\dataloader.py", line 1132, in _try_get_data data = self._data_queue.get(timeout=timeout) File "E:\softwareyang\envs\pytorch_env3.10\lib\multiprocessing\queues.py", line 114, in get raise Empty _queue.Empty

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "F:\jinpengjuGrade22master\ExposureCorrection-main\test.py", line 41, in for i, data in enumerate(dataset): File "E:\softwareyang\envs\pytorch_env3.10\lib\site-packages\torch\utils\data\dataloader.py", line 630, in next data = self._next_data() File "E:\softwareyang\envs\pytorch_env3.10\lib\site-packages\torch\utils\data\dataloader.py", line 1328, in _next_data idx, data = self._get_data() File "E:\softwareyang\envs\pytorch_env3.10\lib\site-packages\torch\utils\data\dataloader.py", line 1294, in _get_data success, data = self._try_get_data() File "E:\softwareyang\envs\pytorch_env3.10\lib\site-packages\torch\utils\data\dataloader.py", line 1145, in _try_get_data raise RuntimeError(f'DataLoader worker (pid(s) {pids_str}) exited unexpectedly') from e RuntimeError: DataLoader worker (pid(s) 19580) exited unexpectedly

how to fix it?

thanks!!!