update README mentioning --ultrasmallnet

javiribera · Oct 2, 2019 · 39b684f · 39b684f
1 parent cb3f966
commit 39b684f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -83,6 +83,8 @@ Models are trained separately for each of the four datasets, as described in the
 
 The [COPYRIGHT](COPYRIGHT.txt) of the pre-trained models is the same as in this repository.
 
+As described in the paper, the pre-trained model for the pupil dataset excludes the five central layers. Thus if you want to use this model you will have to use the option `--ultrasmallnet`.
+
 ## Uninstall
 <pre>
 conda deactivate object-locator