Skip to content

Commit

Permalink
first commit
Browse files Browse the repository at this point in the history
  • Loading branch information
JasonMa2016 committed May 24, 2023
1 parent 8cb201a commit 571c0b6
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ For EpicKitchen or equivalent large-scale pre-training, we suggest using config
```
python train_liv.py training=pretrain dataset=epickitchen
```
Each training run will generate a run folder under ``train_liv_realrobot`` and the reward curves for intermediate model snapshots will be saved in ``\reward_curves`` of the run folder.
Each training run will generate a training run folder under ``train_liv_realrobot`` and the reward curves for intermediate model snapshots will be saved in ``\reward_curves`` of the run folder.

## Multi-Modal Reward Curve Generation
We can use the same training code to also only generate the (animated) reward curves by setting ``eval=True``
Expand All @@ -119,6 +119,10 @@ We can also specify a model path (e.g., ``snapshot.pt`` saved in a run folder) a
```
python train_liv.py eval=True load_snap=PATH_TO_LIV_MODEL dataset=realrobot animate=True
```
In the run folder, you should see animated reward curves like the following:
<p float="left">
<img src="liv/assets/liv_finetuned_appleinblackpot.gif" width="800">
</p>

## License

Expand Down

0 comments on commit 571c0b6

Please sign in to comment.