`model` becames unpickable after calling `trainer.predict(model, dataloader)`

[randommm]

🐛 Bug

model becames unpickable after calling trainer.predict(model, dataloader), raises:
TypeError: can't pickle _thread.lock objects.

See attached example.

example_pickle_pl.zip

[justusschock]

Hi @randommm ,

Can you give us a minimal example to reproduce? Without that, I'm afraid we cannot do much.

[carmocca]

@justusschock he did provide code.

See attached example.

example_pickle_pl.zip

[yifuwang]

Hi @justusschock, in case you haven't had a chance to look at this yet: #7608.

[justusschock]

Hi @yifuwang I just prepared #7609 :D

[yifuwang]

@justusschock oops apologize for the trouble. Should've asked first. Will be more mindful next time :)