UmURL/script_pretrain.sh at main · HuiGuanLab/UmURL · GitHub

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
mkdir -p ./checkpoints/ntu60_xsub
# mkdir -p ./checkpoints/ntu60_xview
# mkdir -p ./checkpoints/ntu120_xsub
# mkdir -p ./checkpoints/ntu120_xset

CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun  --nproc_per_node=4 pretrain.py   \
  --lr 0.0005   --batch-size 512  \
  --checkpoint-path ./checkpoints/ntu60_xsub \
  --schedule 351  --epochs 451  \
  --pre-dataset ntu60  --protocol cross_subject | tee -a ./checkpoints/ntu60_xsub/train.log

# CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun  --nproc_per_node=4 pretrain.py   \
#   --lr 0.0005   --batch-size 1024  \
#   --checkpoint-path ./checkpoints/ntu60_xview \
#   --schedule 351  --epochs 451  \
#   --pre-dataset ntu60  --protocol cross_view | tee -a ./checkpoints/ntu60_xview/train.log

# CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun  --nproc_per_node=4 pretrain.py   \
#   --lr 0.0005   --batch-size 1024  \
#   --checkpoint-path ./checkpoints/ntu120_xsub \
#   --schedule 351  --epochs 451  \
#   --pre-dataset ntu120  --protocol cross_subject | tee -a ./checkpoints/ntu120_xsub/train.log

# CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun  --nproc_per_node=4 pretrain.py   \
#   --lr 0.0005   --batch-size 1024  \
#   --checkpoint-path ./checkpoints/ntu120_xset \
#   --schedule 351  --epochs 451  \
#   --pre-dataset ntu120  --protocol cross_setup | tee -a ./checkpoints/ntu120_xset/train.log