train_wiki20m.sh

#!/bin/bash
export PYTHONUNBUFFERED=1
pretrain_path='bert-base-uncased'
train_file='benchmark/wiki20m/wiki20m_train.txt'
val_file='benchmark/wiki20m/wiki20m_val.txt'
test_file='benchmark/wiki20m/wiki20m_test.txt'
rel2id_file='benchmark/wiki20m/wiki20m_rel2id.json'
batch_size=32
max_length=512
max_epoch=5
seed=772
name=$pretrain_path'_'$max_length'_'$batch_size'_'$max_epoch'_'$seed'_wiki20m'
echo $name
python main.py --pretrain_path $pretrain_path --train_file $train_file --val_file $val_file --test_file $test_file --rel2id_file $rel2id_file --batch_size $batch_size --max_epoch $max_epoch --max_length $max_length --ckpt $name --devs 0