| /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 | |