allow passing dataset to `train_model`

understanding-search / maze-transformer

This repo is built to facilitate the training and analysis of autoregressive transformers on maze-solving tasks.

24 stars 6 forks source link

allow passing dataset to `train_model` #194

Open mivanit opened 1 year ago

mivanit commented 1 year ago

currently in the training notebook, we first create the dataset, and then load it again because we only pass the ConfigHolder to train_model. There should be an option to pass the dataset, and then we check the dataset config for correctness (or, if overridden, modify the ConfigHolder)