updated dependencies

plkmo · Mar 2, 2020 · df01c57 · df01c57
1 parent 6265e93
commit df01c57
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -23,7 +23,7 @@ The purpose of this toolkit is to allow for **easy training/inference of state-o
 ---
 
 ## Pre-requisites
-torch==1.2.0 ; spacy==2.1.8 ; torchtext==0.4.0 ; seqeval==0.0.12 ; pytorch-nlp==0.4.1 
+torch==1.4.0 ; spacy==2.1.8 ; torchtext==0.4.0 ; seqeval==0.0.12 ; pytorch-nlp==0.4.1 
 For mixed precision training (-fp16=1), apex must be installed: [apex==0.1](https://github.com/NVIDIA/apex) 
 For chinese support in Translation: jieba==0.39 
 For ASR: librosa==0.7.0 ; soundfile==0.10.2 

diff --git a/classify.py b/classify.py
@@ -42,7 +42,7 @@
  parser.add_argument('--nb_heads', type=int, default=8, help='Number of head attentions for GAT')
  parser.add_argument("--tokens_length", type=int, default=200, help="Max tokens length for BERT")
  parser.add_argument("--num_classes", type=int, default=66, help="Number of prediction classes (starts from integer 0)")
- parser.add_argument("--train_test_split", type=int, default=0, help="0: No, 1: Yes (Only activate if infer.csv contains labelled data)")
+ parser.add_argument("--train_test_split", type=int, default=1, help="0: No, 1: Yes (Only activate if infer.csv contains labelled data)")
  parser.add_argument("--test_ratio", type=float, default=0.1, help="GCN: Ratio of test to training nodes")
  parser.add_argument("--batch_size", type=int, default=32, help="Training batch size")
  parser.add_argument("--gradient_acc_steps", type=int, default=2, help="No. of steps of gradient accumulation")
@@ -58,7 +58,7 @@
  \n5: XLMRoBERTa''')
 
  parser.add_argument("--train", type=int, default=1, help="Train model on dataset")
- parser.add_argument("--infer", type=int, default=0, help="Infer input sentence labels from trained model")
+ parser.add_argument("--infer", type=int, default=1, help="Infer input sentence labels from trained model")
  args = parser.parse_args()
  save_as_pickle("args.pkl", args)
 

diff --git a/requirements.txt b/requirements.txt
@@ -13,8 +13,8 @@ sentencepiece==0.1.83
 seqeval==0.0.12
 six==1.12.0
 spacy==2.1.8
-torch==1.2.0
+torch==1.4.0
 torchaudio==0.3.0
 torchtext==0.4.0
-torchvision==0.4.0a0+6b959ee
+torchvision==0.5.0
 tqdm==4.32.1
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
  name="nlptoolkit",
- version="0.0.9",
+ version="0.0.10",
  author="Soh Wee Tee",
  author_email="[email protected]",
  description="NLP toolkit",