Skip to content

Rohan alternate rnn analysis#7

Open
RohanKoshy wants to merge 3 commits intomainfrom
Rohan_alternateRNN_analysis
Open

Rohan alternate rnn analysis#7
RohanKoshy wants to merge 3 commits intomainfrom
Rohan_alternateRNN_analysis

Conversation

@RohanKoshy
Copy link
Collaborator

Alternate RNN code which produces rough square-like grid cells as well as a basic analysis notebook to see gridness.

Important note: Change the target directory for storing trained data to personalized location in updated_generate_data.py. ex: /mldata/yourname/trajectories.
This will also need to be done in the so2_gru_glayer_updated.yaml for the data_path and checkpoint_directory.

Generate Data:
python scripts/updated_generate_data.py --manifold so2 --pc_layout product --pc_kernel dos --kappa_excite 6.0 --kappa_inhib 3.0 --n_place_cells 256 --seq_length 500 --n_train 50000 --n_val 5000

Train:
python scripts/train_updated.py --config articulated/configs/estimation/gru_so2_glayer_updated.yaml

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants