Skip to content

Commit

Permalink
updated NER deleted extra folder
Browse files Browse the repository at this point in the history
  • Loading branch information
plkmo committed Sep 20, 2019
1 parent d53de65 commit 9f344ff
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ner.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
parser.add_argument("--train_path", type=str, default="./data/ner/conll2003/eng.train.txt", help="Path to training data txt file")
parser.add_argument("--test_path", type=str, default="./data/ner/conll2003/eng.testa.txt", help="Path to test data txt file (if any)")
parser.add_argument("--num_classes", type=int, default=9, help="Number of prediction classes (starts from integer 0)")
parser.add_argument("--batch_size", type=int, default=4, help="Batch size")
parser.add_argument("--batch_size", type=int, default=8, help="Batch size")
parser.add_argument("--tokens_length", type=int, default=128, help="Max tokens length for BERT")
parser.add_argument("--max_steps", default=-1, type=int,
help="If > 0: set total number of training steps to perform. Override num_train_epochs.")
Expand All @@ -31,7 +31,7 @@
help="Weight decay if we apply some.")
parser.add_argument("--adam_epsilon", default=1e-8, type=float,
help="Epsilon for Adam optimizer.")
parser.add_argument("--gradient_acc_steps", type=int, default=6, help="No. of steps of gradient accumulation")
parser.add_argument("--gradient_acc_steps", type=int, default=1, help="No. of steps of gradient accumulation")
parser.add_argument("--max_norm", type=float, default=1.0, help="Clipped gradient norm")
parser.add_argument("--num_epochs", type=int, default=3, help="No of epochs")
parser.add_argument("--lr", type=float, default=5e-5, help="learning rate")
Expand Down
Binary file not shown.

0 comments on commit 9f344ff

Please sign in to comment.