-
Notifications
You must be signed in to change notification settings - Fork 5
/
pointbert-mn40-32v-middle-mlp_head-4s-1.sh
executable file
·67 lines (56 loc) · 1.38 KB
/
pointbert-mn40-32v-middle-mlp_head-4s-1.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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# 2023/09/30
# NOTE
# 1. ulip2
# 2. class pos: front -> middle
# 3. head_type: ppt-ffn
main_program=main_fewshot.py
proj_name=fewshot
exp_name=pointbert-mn40-32v-middle-mlp_head-4s-1
task=fewshot
model=ULIP_PointBERT
head_type=2 # ppt-ffn
context_len=32
class_pos=middle
dataset=modelnet40_fs
npoints=1024
nshots=4
optim=adamw
lr=0.003
smooth=0.2
gpu=1
epochs=250
batch_size=40
print_freq=60
output_dir=outputs
log_file=run.log
if [ ! -d ${output_dir}/${proj_name}/${exp_name} ]
then
mkdir -p ${output_dir}/${proj_name}/${exp_name}
fi
if [ ! -f ${output_dir}/${proj_name}/${exp_name}/${log_file} ]
then
touch ${output_dir}/${proj_name}/${exp_name}/${log_file}
fi
pueue add -g ${proj_name} \
python ${main_program} \
--proj_name ${proj_name} \
--exp_name ${exp_name} \
--main_program ${main_program} \
--task ${task} \
--model ${model} --ulip2 \
--head_type ${head_type} \
--num_learnable_prompt_tokens ${context_len} \
--class_name_position ${class_pos} \
--dataset_name ${dataset} \
--npoints ${npoints} \
--nshots ${nshots} \
--optim ${optim} \
--lr ${lr} \
--label_smoothing ${smooth} \
--gpu ${gpu} \
--epochs ${epochs} \
--batch_size ${batch_size} \
--print_freq ${print_freq} \
--output_dir ${output_dir} \
--wandb \
2>&1 | tee ${output_dir}/${proj_name}/${exp_name}/${log_file}