Train model on trajectory data

bcnf_traj/config.json

@@ -0,0 +1 @@
+{"data_kwargs": {"T": 2, "dt": 0.06666666666666667, "dataset_name": "test_resimulation_data_T2_dt_1_15", "parameter_selection": ["x0_x", "x0_y", "x0_z", "v0_x", "v0_y", "v0_z", "g_z", "w_x", "w_y", "w_z", "b", "A", "Cd", "rho", "m", "a_x", "a_y", "a_z", "r"], "steps": 30}, "optimizer_kwargs": {"lr": 0.0002}, "lr_scheduler_kwargs": {"mode": "min", "factor": 0.5, "patience": 250, "threshold_mode": "abs", "threshold": 0.1}, "model_kwargs": {"size": 19, "condition_size": 80, "model_nested_size": 16, "model_nested_layers": 7, "model_n_blocks": 32, "model_act_norm": true, "model_dropout": 0.383, "feature_network_input_:size": 30, "feature_network_hidden_size": 59, "feature_network_hidden_layers": 0, "feature_network_dropout": 0.244}}

bcnf_traj/loss_history.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f26bee1e32c64531bd1cba1eb24210c0b4e0b14e60df9dc3d505346a30f099e0
+size 2197732

bcnf_traj/state_dict.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21ea01140203b24b8d5e78150d61502fafeaa4e19cec8a5d5cdf463daff6e507
+size 682750