moreo
/
kernel_authorship
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

adding patience as an argument

This commit is contained in:
Alejandro Moreo Fernandez 2021-02-15 17:58:58 +01:00
parent 79f475e502
commit 1ff1f17039
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main.py
View File

@ -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)