Skip to content

Code for Liebe et al. "Phase of firing does not reflect temporal order in sequence memory of humans and recurrent neural networks", code by Matthijs Pals (@Matthijspals)

License

Notifications You must be signed in to change notification settings

mackelab/sequence-memory

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Code for Phase of firing does not reflect temporal order in sequence memory of humans and recurrent neural networks

Link to preprint

Example usage

Install the conda environment and train an RNN model:

cd sequence-memory
conda env create -f sequence.yml
activate sequence
python rnn_model/rnn/run_single_model.py

Expected behavior: A trained RNN model performing a working-memory task should be obtained and saved.

Expected run-time: One single RNN model took on average around 5-6 hours to train on a Nvidea 2080-ti GPU.

Reproducing the paper Figures

Pull the model and data files from this repo, by first installing git lfs. Alternatively, retrain new RNNs using

python rnn_model/rnn/run_single_model.py

or

wandb sweep rnn_model/rnn/sweep.yml
rnn_model/rnn/run_sweep.py # after adding sweep ID to top of file

and obtain summary statistics over multiple models using

rnn_model/rnn/run_summary.py

Either-way, the figures can then be recreated by running the notebooks in: rnn_model/generate_figures.

Note

Code tested on a Ubuntu system with package versions given in the sequence.yml file

About

Code for Liebe et al. "Phase of firing does not reflect temporal order in sequence memory of humans and recurrent neural networks", code by Matthijs Pals (@Matthijspals)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published