Extend crash datapipe, readers, and update README (#1194)

* update license headers- second try

* modularize reader

* configurable features, node features as a dict of tensors

* update readme

* formatting

* add vtp reader, update readme

* add crash animation

* readme reformatting

* update readme

* add roof crash and crushcan results

* readme formatting

* add roof crash and crushcan results

* add roof crash and crushcan results

* readme references

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/README.md

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* Update examples/structural_mechanics/crash/vtp_reader.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* address review comments

* formatting

---------

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

Author: mnabian

docs/img/crash/crash_case4_reduced.gif

@@ -25,6 +25,7 @@ experiment_desc: "unified training recipe for crash models"
 run_desc: "unified training recipe for crash models"
 
 defaults:
+  - reader: vtp #d3plot
   - datapipe: point_cloud   # will be overridden by model configs
   - model: transolver_autoregressive_rollout_training
   - training: default

docs/img/crash/crushcan.gif

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 _target_: datapipe.CrashGraphDataset
+_convert_: all
 data_dir: ${training.raw_data_dir}
 name: crash_train
 split: train
 num_samples: ${training.num_training_samples}
 num_steps: ${training.num_time_steps}
-wall_node_disp_threshold: 1.0
+features: [thickness]

docs/img/crash/roof_crash.gif

@@ -15,7 +15,8 @@
 # limitations under the License.
 
 _target_: datapipe.CrashPointCloudDataset
+_convert_: all
 data_dir: ${training.raw_data_dir}
 num_samples: ${training.num_training_samples}
 num_steps: ${training.num_time_steps}
-wall_node_disp_threshold: 1.0
+features: [thickness]

examples/structural_mechanics/crash/README.md

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetAutoregressiveRolloutTraining
+_convert_: all
 
 input_dim_nodes: 7                 # pos(3) + vel(3) + thickness(1)
 input_dim_edges: 4                 # dx, dy, dz, distance

examples/structural_mechanics/crash/conf/config.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetAutoregressiveRolloutTraining
+_convert_: all
 
 input_dim_nodes: 7                 # pos(3) + vel(3) + thickness(1)
 input_dim_edges: 4                 # dx, dy, dz, distance

examples/structural_mechanics/crash/conf/datapipe/graph.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetTimeConditionalRollout
+_convert_: all
 
 input_dim_nodes: 5                  # pos(3) + thickness(1) + time(1)
 input_dim_edges: 4                  # dx, dy, dz, distance