Skip to content

Latest commit

 

History

History
46 lines (33 loc) · 1.45 KB

README.md

File metadata and controls

46 lines (33 loc) · 1.45 KB

Overview

rslearn_projects contains the training datasets, model weights, and corresponding code for machine learning applications built on top of rslearn at Ai2.

Setup

Install rslearn:

git clone https://github.com/allenai/rslearn.git
cd rslearn
pip install .[extra]

Install requirements:

cd ..
git clone https://github.com/allenai/rslearn_projects.git
cd rslearn_projects
pip install -r requirements.txt

rslearn_projects includes tooling that expects model checkpoints and auxiliary files to be stored in an RSLP_PREFIX directory. Create a file .env to set the RSLP_PREFIX environment variable:

mkdir project_data
echo "RSLP_PREFIX=project_data/" > .env

Applications