File size: 2,090 Bytes
b71de11 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | /user/hominhduy.nguyen01/u24877/.local/lib/python3.10/site-packages/tyro/_parsers.py:379: UserWarning: The field `data.repo-id` is annotated with type `<class 'str'>`, but the default value `None` has type `<class 'NoneType'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
warnings.warn(message)
/user/hominhduy.nguyen01/u24877/.local/lib/python3.10/site-packages/tyro/_parsers.py:379: UserWarning: The field `model.action-expert-variant` is annotated with type `typing.Literal['dummy', 'gemma_300m', 'gemma_2b', 'gemma_2b_lora']`, but the default value `gemma_300m_lora` has type `<class 'str'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
warnings.warn(message)
04:26:49.173 [I] Running on: glogin9 (167422:train.py:195)
Traceback (most recent call last):
File "/mnt/vast-kisski/projects/kisski-spath/VLA_Groot/in_context_learning/ricl_openpi/scripts/train.py", line 273, in <module>
main(_config.cli())
File "/mnt/vast-kisski/projects/kisski-spath/VLA_Groot/in_context_learning/ricl_openpi/scripts/train.py", line 197, in main
if config.batch_size % jax.device_count() != 0:
File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1098, in device_count
return int(get_backend(backend).device_count())
File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1067, in get_backend
return _get_backend_uncached(platform)
File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1046, in _get_backend_uncached
bs = backends()
File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 953, in backends
assert _default_backend is not None
AssertionError
|