forked from amathislab/DeepDraw
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain_all_regression_tasks_joint_pos_vel.sh
67 lines (47 loc) · 2.87 KB
/
train_all_regression_tasks_joint_pos_vel.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
echo "Training model on each dataset"
########## REGRESS
### Shallow TCNs
### If you need to change the target keys:
# Hand related targets: 'endeffector_coords', 'endeffector_vel', 'endeffector_acc'
# Elbow related targets: 'elbow_coords', 'elbow_vel', 'elbow_acc'
# Joint related targets: 'joint_coords', 'joint_vel', 'joint_acc'
### The following in an example for training TCNs to predict joint position and velocity.
for j in {0..50..1} #20
do
echo "Attempt $i "
echo "Index $j "
let "end_idx=$j+1"
## exp_id 17416
python3.6 train_all_regression_tasks.py --type conv --arch_type spatial_temporal \
--exp_id 616 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
python3.6 train_all_regression_tasks.py --type conv --arch_type temporal_spatial \
--exp_id 616 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
python3.6 train_all_regression_tasks.py --type conv --arch_type spatiotemporal \
--exp_id 616 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
done
### Deeper TCNs
for j in {0..50..1} #20
do
echo "Attempt $i "
echo "Index $j "
let "end_idx=$j+1"
# exp_id 17431
python3.6 train_all_regression_tasks.py --type conv_new --arch_type spatial_temporal \
--exp_id 631 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
python3.6 train_all_regression_tasks.py --type conv_new --arch_type temporal_spatial \
--exp_id 631 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
python3.6 train_all_regression_tasks.py --type conv_new --arch_type spatiotemporal \
--exp_id 631 --start_id $j --end_id $end_idx \
--task regress_joints_pos_vel \
--target_keys joint_coords joint_vel
done