when i run on windows,it loop argument must agree with lock

[cyq123-git]

import PIL import jax.numpy as jnp from tensorflow.io import gfile from scenic.projects.boundary_attention.configs import base_config from scenic.projects.boundary_attention.helpers import train_utils from scenic.projects.boundary_attention.helpers import viz_utils

######## MODIFY THE OPTIONS BELOW #########

im_height = 216 # Replace with height to resize input to im_width = 216 # Replace with width to resize input to

img_path = './pic/noisy_flower.png' # Replace with path to new input weights_dir = './pretrained_weights/' # Add path to pretrained weights here

############################################

input_img = jnp.array(PIL.Image.open(gfile.GFile(img_path, 'rb')).resize((im_width, im_height)))/255.0

print(input_img)

input_img = jnp.expand_dims(input_img.transpose(2,0,1)[:3,:,:], axis=0)

config = base_config.get_config(model_name='boundary_attention', dataset_name='testing', input_size=(im_height, im_width, 3))

apply_jitted, trained_params = train_utils.make_apply(config, weights_dir)

outputs = apply_jitted(trained_params['params'], input_img) viz_utils.visualize_outputs(input_img, outputs)

Traceback (most recent call last): File "D:\python_software\code_doc\2023project\googleboundary\test.py", line 27, in apply_jitted, trained_params = train_utils.make_apply(config, weights_dir) File "D:\python_software\anaconda\envs\google\lib\site-packages\scenic\projects\boundary_attention\helpers\train_utils.py", line 36, in make_apply trained_params = checkpoints.restore_checkpoint(ckpt_dir=ckpt_dir, File "D:\python_software\anaconda\envs\google\lib\site-packages\flax\training\checkpoints.py", line 1129, in restore_checkpoint restored = orbax_checkpointer.restore( File "D:\python_software\anaconda\envs\google\lib\site-packages\orbax\checkpoint\checkpointer.py", line 170, in restore restored = self._handler.restore(directory, args=ckpt_args) File "D:\python_software\anaconda\envs\google\lib\site-packages\orbax\checkpoint\pytree_checkpoint_handler.py", line 1088, in restore byte_limiter = get_byte_limiter(self._concurrent_gb) File "D:\python_software\anaconda\envs\google\lib\site-packages\orbax\checkpoint\pytree_checkpoint_handler.py", line 171, in get_byte_limiter return asyncio.run(_create_byte_limiter()) File "D:\python_software\anaconda\envs\google\lib\asyncio\runners.py", line 44, in run return loop.run_until_complete(main) File "D:\python_software\anaconda\envs\google\lib\asyncio\base_events.py", line 641, in run_until_complete return future.result() File "D:\python_software\anaconda\envs\google\lib\site-packages\orbax\checkpoint\pytree_checkpoint_handler.py", line 169, in _create_byte_limiter return LimitInFlightBytes(concurrent_bytes) # pylint: disable=protected-access File "D:\python_software\anaconda\envs\google\lib\site-packages\jax\experimental\array_serialization\serialization.py", line 180, in init self._cv = asyncio.Condition(lock=asyncio.Lock()) File "D:\python_software\anaconda\envs\google\lib\asyncio\locks.py", line 234, in init raise ValueError("loop argument must agree with lock") ValueError: loop argument must agree with lock

Process finished with exit code 1