-
Notifications
You must be signed in to change notification settings - Fork 1
/
do_all_cf_zeros_random_mean.sh
executable file
·24 lines (22 loc) · 2.07 KB
/
do_all_cf_zeros_random_mean.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
#!/bin/bash
dataset=$1
model=$2
if [ "$model" = "bprmf" ]; then
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_${model}.py --dataset ${dataset} > ${model}_${dataset}.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model vbpr --method zeros > vbpr_${dataset}_zeros.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model vbpr --method random > vbpr_${dataset}_random.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model vbpr --method mean > vbpr_${dataset}_mean.out 2>&1
elif [ "$model" = "freedom" ]; then
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model freedom --method zeros > freedom_${dataset}_zeros.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model freedom --method random > freedom_${dataset}_random.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model freedom --method mean > freedom_${dataset}_mean.out 2>&1
elif [ "$model" = "bm3" ]; then
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model bm3 --method zeros > bm3_${dataset}_zeros.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model bm3 --method random > bm3_${dataset}_random.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model bm3 --method mean > bm3_${dataset}_mean.out 2>&1
else
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_${model}.py --dataset ${dataset} > ${model}_${dataset}.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model ${model}m --method zeros > ${model}m_${dataset}_zeros.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model ${model}m --method random > ${model}m_${dataset}_random.out 2>&1
CUBLAS_WORKSPACE_CONFIG=:4096:8 python run_multimodal.py --dataset ${dataset} --model ${model}m --method mean > ${model}m_${dataset}_mean.out 2>&1
fi