Dataset

README.md

@@ -20,8 +20,9 @@ repository](http://github.com/alignmentresearch/learned-planner), and possibly t
 
 # Model details
 
-**Hyperparameters:** see `model/*/cp_*/cfg.json` for the hyperparameters that were used to train a particular run.
+## Hyperparameters:
 
+See `model/*/cp_*/cfg.json` for the hyperparameters that were used to train a particular run.
 
 ## Parameter counts:
 
@@ -29,6 +30,10 @@ repository](http://github.com/alignmentresearch/learned-planner), and possibly t
 - DRC(1, 1):  987,525 (0.99M)
 - ResNet:  3,068,421 (3.07M)
 
+## Training dataset:
+
+The [Boxoban set of levels by DeepMind](https://github.com/google-deepmind/boxoban-levels).
+
 # Citation
 
 If you use these neural networks, please cite our work: