diff --git a/src/main.py b/src/main.py
index 897a34a..18e420c 100644
--- a/src/main.py
+++ b/src/main.py
@@ -209,6 +209,7 @@ if __name__ == '__main__':
     parser.add_argument('-r', '--repr', help='Projection size (phi)', type=int, default=256)
     parser.add_argument('-k', '--kernelsizes', help='Size of the convolutional kernels', nargs='+', default=[6,7,8])
     parser.add_argument('-p', '--pad', help='Pad length', type=int, default=3000)
+    parser.add_argument('--patience', help='Patience limit', type=int, default=20)
     parser.add_argument('-b', '--batchsize', help='Batch size', type=int, default=100)
     parser.add_argument('-e', '--epochs', help='Max number of epochs', type=int, default=500)
     parser.add_argument('-A', '--authors', help='Number of authors (-1 to select all)', type=int, default=-1)