This folder contains the code for LaGR-SEQ in cube stacking, image completion and object arrangement (corresponding to extended results shown in the appendix). The python code for each environment is found within the corresponding directories. Each directory contains:
- Python code for LaGR-SEQ
- Python code for LaGR without SEQ
- Python code for DQN
- Python code (including text descriptor) for querying LLM from server
- LLM cache file
The environments considered are:
- Cube stacking
- Image completion
- Object arrangement (extended versions of the simulated robot experiments)
- LaGR-SEQ
- LaGR without SEQ
- DQN (Mnih et al.,"Human-level control through deep reinforcement learning") or Q learning for cube stacking
conda create --name lagrseq python=3.9
conda activate lagrseq
Install dependencies with pip install -r requirements.txt
Users will also need a paid account with a valid key to access OpenAI API
Set openai API key using instructions from: https://help.openai.com/en/articles/5112595-best-practices-for-api-key-safety
##Sample call: python LaGRSEQ.py