forked from Walter0807/MotionBERT
-
Notifications
You must be signed in to change notification settings - Fork 1
/
slurm_infer.sh
75 lines (63 loc) · 2.33 KB
/
slurm_infer.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
68
69
70
71
72
73
74
75
#!/bin/bash -l
#SBATCH --job-name=MB_infer
#SBATCH --output=output_slurm/infer_log.txt
#SBATCH --error=output_slurm/infer_error.txt
#SBATCH --mail-type=BEGIN,END,FAIL
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=2
#SBATCH --mem=20g
#SBATCH --gres=gpu:1
#SBATCH --partition=gpu
##SBATCH --partition=debug
#SBATCH --time=00:30:00
#SBATCH --account=shdpm0
##### END preamble
##### Run in MotionBert dir
my_job_header
module load python3.10-anaconda
module load cuda/11.8.0
module load cudnn/11.8-v8.7.0
module load cupti/11.8.0
module load python/3.10.4
module load pytorch/2.0.1
module list
#conda activate motionbert
echo "cpu-2, gpu-1, mem-20"
# Dataset
#config_file="configs/pose3d/MB_train_h36m.yaml"
#config_file="configs/pose3d/MB_ft_h36m.yaml"
#config_file="configs/pose3d/MB_train_VEHSR3.yaml"
#config_file="configs/pose3d/MB_ft_VEHSR3_6DPose.yaml"
#config_file="configs/pose3d/RTMPose_exp/MB_ft_VEHS_config6.yaml"
#config_file="configs/pose3d/RTMPose_exp/MB_ft_inference.yaml"
#config_file="configs/pose3d/RTMPose_exp/MB_ft_VEHS_tilt_correct.yaml"
#config_file="configs/pose3d/hand/MB_train_Rokoko.yaml"
config_file="configs/pose3d/hand/MB_infer_lab_RTMinput.yaml"
# Checkpoint
#checkpoint_bin="checkpoint/pose3d/MB_train_h36m/best_epoch.bin"
#checkpoint_bin="checkpoint/pose3d/MB_ft_VEHSR3_6DPose/best_epoch.bin"
#checkpoint_bin="checkpoint/pose3d/FT_MB_release_MB_ft_h36m/best_epoch.bin"
#checkpoint_bin="checkpoint/pose3d/FT_RTM_VEHS_config6_GT2d_true/best_epoch.bin"
#checkpoint_bin="checkpoint/pose3d/FT_RTM_VEHS_config6/best_epoch.bin"
#checkpoint_bin="checkpoint/pose3d/FT_RTM_VEHS_tilt_corrected/best_epoch.bin"
checkpoint_bin="checkpoint/pose3d/MB_train_Rokoko_hand_21/latest_epoch.bin"
echo "config_file: $config_file"
echo "checkpoint_bin: $checkpoint_bin"
## inference only
#python -u infer3d_train.py \
#--config "$config_file" \
#--wandb_project "MotionBert_eval" \
#--wandb_name "RTM_input_MB_ft_inference_industry_3" \
#--note "save pose output" \
#--out_path "experiment/RTM2D_ft/industry_3_RTM2d" \
#--test_set_keyword validate \
#--evaluate "$checkpoint_bin"
#> "output_slurm/eval_${SLURM_JOB_ID}_output.out"
# (config change 4 location)
# Hand-21
python -u infer3d_train.py \
--config "$config_file" \
--out_path "experiment/handPose/Lab_RTMin_wRokoko_2/right" \
--test_set_keyword test \
--evaluate "$checkpoint_bin"