Skip to content

Commit

Permalink
Updated readme file
Browse files Browse the repository at this point in the history
  • Loading branch information
AntreasAntoniou committed Apr 22, 2018
1 parent fdb5618 commit 4271376
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,6 @@ python train_one_shot_learning_matching_network.py --batch_size 32 --experiment_
```

And then run `python train_one_shot_learning_matching_network.py`

## Features
The code supports automatic checkpointing as well as statistics saving.
It uses 1200 classes for training, 211 classes for testing and 211 classes for validation. We save the latest 5 trained models as well as keep track
Expand All @@ -69,6 +67,9 @@ Our implementation uses the omniglot dataset, but one can easily add a new data
experiment by passing the data provider to the ExperimentBuilder class and the system should work with it, as long as
it provides the batches in the same way as our data provider, more details can be found in data.py

We've also very recently introduced the Full Context Embeddings version of matching networks properly implemented as
explained in the paper.

## Acknowledgements
Special thanks to https://github.com/zergylord for his Matching Networks
implementation of which parts were used for this implementation. More
Expand Down

0 comments on commit 4271376

Please sign in to comment.