forked from TudorAvarvarei/LNN_behavioural_cloning_quadrotor
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_config.yaml
More file actions
25 lines (21 loc) · 759 Bytes
/
Copy pathtest_config.yaml
File metadata and controls
25 lines (21 loc) · 759 Bytes
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
model_path: "mlp_epoch=19_val_loss=0.003130.ckpt" # Saved checkpoint/config stem from training
dataset: # Test dataset selection
desired_trajectories: 1
test_path: datasets/hover_dataset_test.npz
trajectory_plotter: false # Plot the first trajectory after evaluation
sensitivity_analysis: true # Evaluate the whole test set instead of a single random subset
dataloader: # Test-time loader settings
batch_size: 1
num_workers: 8
pin_memory: false
drop_last: true
ablation: # Automated feature ablation suite
enabled: true
fill_value: 0.0
feature_sets:
position: [dx, dy, dz]
velocity: [vx, vy, vz]
attitude: [phi, theta, psi]
angular_rate: [p, q, r]
disturbance: [Mx_ext, My_ext, Mz_ext]
rotor_speed: [omega]