Skip to content

Commit

Permalink
Change dataset path
Browse files Browse the repository at this point in the history
  • Loading branch information
rdspring1 committed Jul 26, 2017
1 parent 7969a9d commit 90da4f6
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,6 @@ Run data_utils.py to download and pre-process Omniglot dataset
2. [Learning to Remember Rare Events](https://arxiv.org/abs/1703.03129)

# Other References
1. [One-shot Learning with Memory-Augmented Neural Networks](https://arxiv.org/pdf/1605.06065.pdf)
2. [Matching Networks for One Shot Learning](https://arxiv.org/pdf/1606.04080.pdf)
1. [One-shot Learning with Memory-Augmented Neural Networks](https://arxiv.org/abs/1605.06065.pdf)
2. [Matching Networks for One Shot Learning](https://arxiv.org/abs/1606.04080.pdf)
* [Andrej Karpathy's Notes - Matching Networks](https://github.com/karpathy/paper-notes/blob/master/matching_networks.md)
2 changes: 1 addition & 1 deletion cnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def forward(self, x, predict):
episode_length = 30
episode_width = 5
validation_frequency = 20
DATA_FILE_FORMAT = os.path.join(os.getcwd(), 'datasets/omniglot', '%s_omni.pkl')
DATA_FILE_FORMAT = os.path.join(os.getcwd(), '%s_omni.pkl')

train_filepath = DATA_FILE_FORMAT % 'train'
trainset = omniglot.OmniglotDataset(train_filepath)
Expand Down

0 comments on commit 90da4f6

Please sign in to comment.