Skip to content

Commit 90da4f6

Browse files
committed
Change dataset path
1 parent 7969a9d commit 90da4f6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,6 @@ Run data_utils.py to download and pre-process Omniglot dataset
88
2. [Learning to Remember Rare Events](https://arxiv.org/abs/1703.03129)
99

1010
# Other References
11-
1. [One-shot Learning with Memory-Augmented Neural Networks](https://arxiv.org/pdf/1605.06065.pdf)
12-
2. [Matching Networks for One Shot Learning](https://arxiv.org/pdf/1606.04080.pdf)
11+
1. [One-shot Learning with Memory-Augmented Neural Networks](https://arxiv.org/abs/1605.06065.pdf)
12+
2. [Matching Networks for One Shot Learning](https://arxiv.org/abs/1606.04080.pdf)
1313
* [Andrej Karpathy's Notes - Matching Networks](https://github.com/karpathy/paper-notes/blob/master/matching_networks.md)

cnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def forward(self, x, predict):
5050
episode_length = 30
5151
episode_width = 5
5252
validation_frequency = 20
53-
DATA_FILE_FORMAT = os.path.join(os.getcwd(), 'datasets/omniglot', '%s_omni.pkl')
53+
DATA_FILE_FORMAT = os.path.join(os.getcwd(), '%s_omni.pkl')
5454

5555
train_filepath = DATA_FILE_FORMAT % 'train'
5656
trainset = omniglot.OmniglotDataset(train_filepath)

0 commit comments

Comments
 (0)