/usr/lib/python3.10/multiprocessing/popen_fork.py:66: 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()
Test: [0/5] Time 0.492 ( 0.492) PSNR 32.21 (32.21) SSIM 0.8546 (0.8546)
PSNR 30.6700 SSIM 0.8626
Its not iterating, i guess.(but training works fine)
when i run test.py, this is the result i get:
/usr/lib/python3.10/multiprocessing/popen_fork.py:66: 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() Test: [0/5] Time 0.492 ( 0.492) PSNR 32.21 (32.21) SSIM 0.8546 (0.8546)
Its not iterating, i guess.(but training works fine)