Can't pickle local object 'linear_scheduler.<locals>.lr_lambda'

[WuChannn]

trafficstars

when i use multiple gpus to train GeoLayoutLM, error as follows:

Can't pickle local object 'linear_scheduler..lr_lambda'

@wdp-007 @congyao @alibaba-oss Looking forward to your reply.

[hossein-pdftron]

Any update on this issue?

[kenneys-bot]

I had the same problem today when I deployed the project to train on multiple GPUs： LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2] File "/public/home/lab70432/workspace/env_sources/geolayoutlm/train.py", line 64, in <module> main() File "/public/home/lab70432/workspace/env_sources/geolayoutlm/train.py", line 60, in main trainer.fit(pl_module, datamodule=data_module) AttributeError: Can't pickle local object 'linear_scheduler.<locals>.lr_lambda' I don't have any solution to this problem yet.

Any update on this issue?

Have you solved the problem yet? Thanks.