Skip to content

Commit

Permalink
Doc update + bug fix in data path constant
Browse files Browse the repository at this point in the history
  • Loading branch information
jordis-ai2 committed Jul 14, 2022
1 parent 0d01f8e commit 73b92a6
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
11 changes: 6 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -572,23 +572,24 @@ evaluation doesn't have to be run on each set separately.

### 🏘 ProcTHOR pre-training

Here we describe commands that can be used to generate a ProcTHOR-pretrained agent and
We include commands that can be used to generate a ProcTHOR-pretrained agent and
then fine-tune it with the 2022 rearrange dataset. Please note that this only covers the 1-phase modality,
for which we also provide a
[pre-trained and fine-tune checkpoint](https://prior-model-weights.s3.us-east-2.amazonaws.com/embodied-ai/rearrangement/one-phase/exp_iThorOnePhaseRGBClipResNet50FineTune_procthor180Msteps_ithor_splits_ithor_fine_tune_64_to_128_rollout_3Msteps_6Msteps__stage_02__steps_000016018675.pt).
We also provide scripts to generate new ProcTHOR datasets in case you want to try different episodes.
[pre-trained and fine-tuned checkpoint](https://prior-model-weights.s3.us-east-2.amazonaws.com/embodied-ai/rearrangement/one-phase/exp_iThorOnePhaseRGBClipResNet50FineTune_procthor180Msteps_ithor_splits_ithor_fine_tune_64_to_128_rollout_3Msteps_6Msteps__stage_02__steps_000016018675.pt).
We also provide scripts to generate new ProcTHOR datasets in case you want to try new episodes.

#### Pre-train model in ProcTHOR (single machine)
The following will take for about 10-14 days on an 8-GPU machine with 56 CPU cores:
```bash
allenact -b baseline_configs/one_phase/procthor one_phase_rgb_clip_dagger \
-s 12345 --config_kwargs '{"distributed_nodes":1}'
```
We strongly recommend using a larger number of GPUs and computing nodes for this step, if possible.
We **strongly** recommend using a larger number of GPUs and computing nodes for this step.

#### ProcTHOR mini-valid
Run ProcTHOR mini-valid on checkpoints under the CKPT_DIR directory:
Run ProcTHOR mini-valid on checkpoints under a `CKPT_DIR` directory:
```bash
inv make-valid-houses-file
allenact -b baseline_configs/one_phase/procthor/eval eval_minivalid_procthor \
-s 12345 --eval --approx_ckpt_step_interval 5e6 -c CKPT_DIR
```
Expand Down
2 changes: 1 addition & 1 deletion rearrange/procthor_rearrange/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@


STARTER_DATA_DIR = os.path.join(
os.path.abspath(os.path.dirname(Path(__file__))), "../data", "2022procthor",
os.path.abspath(os.path.dirname(Path(__file__))), "../../data", "2022procthor",
)

PROCTHOR_COMMIT_ID = "90eac925dc750818890069e3131f899998dc58b4"
3 changes: 0 additions & 3 deletions rearrange/procthor_rearrange/expert.py
Original file line number Diff line number Diff line change
Expand Up @@ -1369,9 +1369,6 @@ def test(
except:
return

# submodule
sys.path.append("ai2thor_rearrangement")

init_logging("debug")
test(
stage="debug",
Expand Down

0 comments on commit 73b92a6

Please sign in to comment.