Trin LSTM and transformer (not final)

bcnf_traj_bilstm_test_resimulation_data_T2_dt_1_15/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_x", "g_y", "g_z", "w_x", "w_y", "w_z", "b", "m", "a_x", "a_y", "a_z", "r", "A", "Cd", "rho"], "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": 21, "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": null, "time_series_network": "LSTMFeatureNetwork", "time_series_network_input_size": 3, "time_series_network_hidden_size": 16, "time_series_network_num_layers": 1, "time_series_network_dropout": 0.244, "time_series_network_bidirectional": true}}

bcnf_traj_bilstm_test_resimulation_data_T2_dt_1_15/loss_history.pkl

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

bcnf_traj_bilstm_test_resimulation_data_T2_dt_1_15/state_dict.pt

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

bcnf_traj_trf_test_resimulation_data_T2_dt_1_15/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_x", "g_y", "g_z", "w_x", "w_y", "w_z", "b", "m", "a_x", "a_y", "a_z", "r", "A", "Cd", "rho"], "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": 21, "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": null, "time_series_network": "Transformer", "time_series_network_d_model": 3, "time_series_network_num_heads": 3, "time_series_network_d_ff": 128, "time_series_network_num_layers": 5}}

bcnf_traj_trf_test_resimulation_data_T2_dt_1_15/loss_history.pkl

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

bcnf_traj_trf_test_resimulation_data_T2_dt_1_15/state_dict.pt

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