.run_ce.sh 546 字节
Newer Older
R
add ce  
root 已提交
1 2 3 4 5
#!/bin/bash

export MKL_NUM_THREADS=1
export OMP_NUM_THREADS=1

Z
zhengya01 已提交
6
cudaid=${text_matching_on_quora:=0} # use 0-th card as default
R
add ce  
root 已提交
7 8
export CUDA_VISIBLE_DEVICES=$cudaid

Z
zhengya01 已提交
9
FLAGS_benchmark=true  python train_and_evaluate.py --model_name=cdssmNet --config=cdssm_base --enable_ce --epoch_num=5 | python _ce.py
R
add ce  
root 已提交
10

Z
zhengya01 已提交
11
cudaid=${text_matching_on_quora_m:=0,1,2,3} # use 0,1,2,3 card as default
R
add ce  
root 已提交
12 13
export CUDA_VISIBLE_DEVICES=$cudaid

Z
zhengya01 已提交
14
FLAGS_benchmark=true  python train_and_evaluate.py --model_name=cdssmNet --config=cdssm_base --enable_ce --epoch_num=5 | python _ce.py