psaegert commited on
Commit
3993ce9
·
unverified ·
1 Parent(s): 2295542

Train model on no particular trajectory data

Browse files
bcnf_traj/config.json DELETED
@@ -1 +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_test_resimulation_data_T2_dt_1_15/config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"data_kwargs": {"T": 1.5, "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_input_size": 90, "feature_network_hidden_size": 59, "feature_network_hidden_layers": 0, "feature_network_dropout": 0.244, "feature_network_batchnorm": false}}
{bcnf_traj → bcnf_traj_test_resimulation_data_T2_dt_1_15}/loss_history.pkl RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f26bee1e32c64531bd1cba1eb24210c0b4e0b14e60df9dc3d505346a30f099e0
3
- size 2197732
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:421c0f90a3a0ad578ff815ffd06220481f61265635f0f7e7463881351c87caaa
3
+ size 2732386
{bcnf_traj → bcnf_traj_test_resimulation_data_T2_dt_1_15}/state_dict.pt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:21ea01140203b24b8d5e78150d61502fafeaa4e19cec8a5d5cdf463daff6e507
3
- size 682750
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ece08470c36741e82eb50a145a69c0b81f0485e1f9cb1cbc510a7c878be520c
3
+ size 752413