-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.sh
26 lines (23 loc) · 888 Bytes
/
setup.sh
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
#!/usr/bin/env bash
set -e
# Main Requirements
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
pip uninstall dataclasses -y # Compatibility issue with Python > 3.6
# Optional: Set up NLTK packages
if [[ -f punkt.zip ]]; then
mkdir -p /home/admin/nltk_data/tokenizers
cp punkt.zip /home/admin/nltk_data/tokenizers
fi
if [[ -f wordnet.zip ]]; then
mkdir -p /home/admin/nltk_data/corpora
cp wordnet.zip /home/admin/nltk_data/corpora
fi
# Make sample data for quick debugging
unzip -n data.zip -d aste/
cd aste/data/triplet_data
mkdir -p sample
head -n 32 14lap/train.txt >sample/train.txt
head -n 32 14lap/dev.txt >sample/dev.txt
head -n 32 14lap/test.txt >sample/test.txt
cd ../../..