deblurganv2运行时遇到的问题Can‘t pickle local object ‘get_corrupt_function.＜locals＞.process‘ #44

在运行github上deblurganv2的train.py时，遇到如下报错Can't pickle local object 'get_corrupt_function.<locals>.process' #44

原因是因为多线程问题，需要修改train.py中的一行代码

get_dataloader = partial(DataLoader, batch_size=batch_size, num_workers=cpu_count(), shuffle=True, drop_last=True)

 大概在178行左右。

将之改为

get_dataloader = partial(DataLoader, batch_size=batch_size, shuffle=True, drop_last=True)

get_dataloader = partial(DataLoader, batch_size=batch_size, shuffle=True, drop_last=True)

即可完成。