Commit dc3fb797 authored by shauryadNv's avatar shauryadNv Committed by Kelly Guo

Changes robomimic eval script to use CPU instead of GPU (#516)

# Description

Changed robomimic eval script to use CPU instead of GPU.
 
## Type of change

- Bug fix (non-breaking change which fixes an issue)

## Checklist

- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there
parent f75bf64f
......@@ -241,7 +241,7 @@ def main() -> None:
env = gym.make(args_cli.task, cfg=env_cfg).unwrapped
# Acquire device
device = TorchUtils.get_torch_device(try_to_use_cuda=True)
device = TorchUtils.get_torch_device(try_to_use_cuda=False)
# Get model checkpoints
model_checkpoints = [f.name for f in os.scandir(args_cli.input_dir) if f.is_file()]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment