2153] Execution of replica 0 failed: INTERNAL: Could not find the corresponding function
Traceback (most recent call last):
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/src/trident_train.py", line 80, in
train_tasks, valid_tasks, _, learner = setup(
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/src/zoo/trident_utils.py", line 40, in setup
train_tasks = gen_tasks(dataset, root, download=download, mode='train',
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/taskers.py", line 104, in gen_tasks
cfs = Primitives(root, mode, transform=image_transforms,
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/loaders.py", line 138, in init
shots = [load_shot(path, shape) for path in load_paths(concept_path)]
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/loaders.py", line 138, in
shots = [load_shot(path, shape) for path in load_paths(concept_path)]
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/primitives.py", line 31, in load_shot
image = load_image(os.path.join(directory, 'image.png'), shape)
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/primitives.py", line 12, in load_image
image = jax.image.resize(image, (*shape, 3), 'bilinear')
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/site-packages/jax/_src/image/scale.py", line 339, in resize
return _resize(image, core.canonicalize_shape(shape), method, antialias,
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/site-packages/jax/_src/traceback_util.py", line 162, in reraise_with_filtered_traceback
return fun(*args, **kwargs)
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/site-packages/jax/_src/api.py", line 623, in cache_miss
out_flat = call_bind_continuation(execute(*args_flat))
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/site-packages/jax/_src/dispatch.py", line 895, in _execute_compiled
out_flat = compiled.execute(in_flat)
jax._src.traceback_util.UnfilteredStackTrace: jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Could not find the corresponding function
The stack trace below excludes JAX-internal frames.
The preceding is the original exception that occurred, unmodified.
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/runpy.py", line 197, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/src/trident_train.py", line 80, in
train_tasks, valid_tasks, _, learner = setup(
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/src/zoo/trident_utils.py", line 40, in setup
train_tasks = gen_tasks(dataset, root, download=download, mode='train',
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/taskers.py", line 104, in gen_tasks
cfs = Primitives(root, mode, transform=image_transforms,
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/loaders.py", line 138, in init
shots = [load_shot(path, shape) for path in load_paths(concept_path)]
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/loaders.py", line 138, in
shots = [load_shot(path, shape) for path in load_paths(concept_path)]
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/primitives.py", line 31, in load_shot
image = load_image(os.path.join(directory, 'image.png'), shape)
File "/home/dfki.uni-bremen.de/csingh/DFKI/PhysWM/trident_primitives/data/primitives.py", line 12, in load_image
image = jax.image.resize(image, (*shape, 3), 'bilinear')
File "/home/dfki.uni-bremen.de/csingh/anaconda3/lib/python3.9/site-packages/jax/_src/image/scale.py", line 339, in resize
return _resize(image, core.canonicalize_shape(shape), method, antialias,
jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Could not find the corresponding functio