wassname/iris_bigvae
1
0
Fork 0
mirror of https://github.com/wassname/iris_bigvae.git synced 2026-09-09 11:24:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Vincent Micheli
2022-10-03 17:41:55 +02:00
parent a25030cc47
commit bf2d6b1943
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -7
View File
@@ -121,13 +121,7 @@ Pretrained models are available in `pretrained_models/`.
- To start a training run from one of these checkpoints, in the section `initialization` of `config/trainer.yaml`, set `path_to_checkpoint` to the corresponding path, and `load_tokenizer`, `load_world_model`, and `load_actor_critic` to `True`.
- To visualize one of these checkpoints, run the following commands:
```bash
mkdir checkpoints
cp pretrained_models/Breakout.pt checkpoints/last.pt
./scripts/play.sh
```
with optionally the flags described above for `play.sh`.
- To visualize one of these checkpoints, set `train.id` to the corresponding game in `config/env/default.yaml`, create a `checkpoints` directory and copy the checkpoint to `checkpoints/last.pt`. You can then visualize the agent with `./scripts/play.sh` as described above.
## Credits