diff --git a/demo/demo1/data/mrqa/convert.py b/demo/demo1/data/mrqa/convert.py deleted file mode 100644 index 1b793a48f971d6fd8048fae5a59c3be1ed972b1a..0000000000000000000000000000000000000000 --- a/demo/demo1/data/mrqa/convert.py +++ /dev/null @@ -1,18 +0,0 @@ -# coding: utf-8 -f='mrqa-combined.train.raw.json' -import json -a=json.load(open(f)) -a=a['data'] -writer = open('train.json','w') - -for s in a: - p = s['paragraphs'] - assert len(p) == 1 - p = p[0] - q = {} - q['context'] = p['context'] - q['qa_list'] = p['qas'] - writer.write(json.dumps(q)+'\n') - -writer.close() - diff --git a/demo/demo2/data/mrqa/convert.py b/demo/demo2/data/mrqa/convert.py deleted file mode 100644 index 1b793a48f971d6fd8048fae5a59c3be1ed972b1a..0000000000000000000000000000000000000000 --- a/demo/demo2/data/mrqa/convert.py +++ /dev/null @@ -1,18 +0,0 @@ -# coding: utf-8 -f='mrqa-combined.train.raw.json' -import json -a=json.load(open(f)) -a=a['data'] -writer = open('train.json','w') - -for s in a: - p = s['paragraphs'] - assert len(p) == 1 - p = p[0] - q = {} - q['context'] = p['context'] - q['qa_list'] = p['qas'] - writer.write(json.dumps(q)+'\n') - -writer.close() - diff --git a/demo/demo2/run.sh b/demo/demo2/run.sh index 5d0a5a0f5468b5bdb3a067fd40a1cfbd033e201a..9e75bc77f12ebe502ff43d58924d54996e777da4 100755 --- a/demo/demo2/run.sh +++ b/demo/demo2/run.sh @@ -1,3 +1,3 @@ -export CUDA_VISIBLE_DEVICES=0,1 +export CUDA_VISIBLE_DEVICES=0 python run.py