-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrun-train-scorer-finetune.py
50 lines (50 loc) · 1.22 KB
/
run-train-scorer-finetune.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/bin/bash
# source you pyenv or conda or whatever here
cd ~/code/ParlAI/
python examples/train_model.py \
-t convai2:self \
--max-train-time 82079 \
--model parlai_external.agents.seq2seq_on_steroids.seq2seq_on_steroids:SteroidSeq2seqAgent \
--init-model model # PUT INITIAL MODEL HERE, from model.tar.gz
--margin 1 \
--num-rank-cand 5 \
--cand-type current_labels \
--lmweight 0.1 \
-bi True \
-pt True \
--strict-load False \
--rankweight 1 \
--rankhiddensize 512 \
--ranknl 4 \
--ranklossreduce elementwise_mean \
--rank-activation Tanh \
-rc True \
-hs 512 \
-esz 300 \
-att general \
-nl 2 \
-rnn lstm \
-lr 0.001 \
-dr 0.5 \
-clip 0.1 \
-lt enc_dec \
-opt adam \
-emb glove \
-mom 0.9 \
-bs 64 \
-bsrt False \
--tensorboard-log True \
--tensorboard-tag margin,num_rank_cand,lmweight,rankweight,rankhiddensize,ranknl,ranklossreduce,rank_activation \
--tensorboard-comment withadam001 \
--tensorboard-metrics loss,rank_loss,ppl,f1,hits@1,hits@5,hits@10 \
-veps 0.5 \
--validation-metric hits@1 \
--validation-metric-mode max \
--validation-patience 24 \
--save-after-valid True \
--load-from-checkpoint True \
--dict-lower True \
--dict-maxexs -1 \
--dict-include-valid True \
--log-every-n-secs 25 \
--model-file #PATH TO SAVE NEWLY TRAINED FILES