CompRhys / aviary

The Wren sits on its Roost in the Aviary.
MIT License
48 stars 11 forks source link

Include `model_params` in Wrenformer checkpoints #49

Closed janosh closed 2 years ago

janosh commented 2 years ago

Write df_to_in_mem_dataloader() to extract train/test_loader creation from run_wrenformer(). New function be used in multiple places like use_trained_wrenformer_ensemble.py.