RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock. self.pid = os.fork()

[melikaaala]

Hi
I have trained machine learning python code on google colab but I faced this error. Is there any solution for this:
os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock. self.pid = os.fork()

[Delgan]

Can you please share the code that produces this warning?

Loguru does not call os.fork().

[melikaaala]

I use this url: https://github.com/Algolzw/image-restoration-sde

[melikaaala]

hi I use this code: https://github.com/Algolzw/image-restoration-sde

…

On Tue, Apr 16, 2024 at 9:29 AM Delgan ***@***.***> wrote: Can you please share the code that produces this warning? Loguru does not call os.fork(). — Reply to this email directly, view it on GitHub <#1125 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ALIF3BCCZTOHLUSXTG3NVR3Y5S43HAVCNFSM6AAAAABGISM4LSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDANJYGI4DQMRYGY> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[Delgan]

Sorry but there is not enough information for me to help you. Please, try to provide a minimal reproducible example involving usage of Loguru.

Right now, I don't understand how the problem you're facing is related to Loguru. As I said, your error is complaining about usage of os.fork() within JAX framework, but Loguru does not make any use of forking.