/Pre-Trained-Language-Models-for-Interactive-Decision-Making

Pre-Trained Language Models for Interactive Decision-Making [NeurIPS 2022]

Primary LanguagePython

Pre-Trained Language Models for Interactive Decision-Making

Project Page | Paper

This is the official codebase for Pre-Trained Language Models for Interactive Decision-Making.

Setup

Get the VirtualHome Simulator

Download VirtualHome exectuable file (v2.2.5). Please refer to VirtualHome for more details.

The VirtualHome exectuable file we used can be downloaded from here. Put it under ./virtualhome.

Download Checkpoints

Download the checkpoint: LID-Text. Put the checkpoint under ./behavior_cloning/checkpoint.

Inference

Download Virtualhome initial environment for testing: data. Put them under ./data.

For interactive evaluation, run:

sh scripts/inference.sh