jeongwhanchoi / hcmt

Learning Flexible Body Collision Dynamics with Hierarchical Mesh Transformer, Accepted in ICLR 2024

HCMT: Learning Flexible Body Collision Dynamics with Hierarchical Mesh Transformer

Set environment

The environment can be set up using either environment.yml file or manually installing the dependencies.

Using an environment.yml file

conda env create -f environment.yml

Requirements

• tensorflow-gpu==2.8.0
• dm-sonnet==2.0.1
• protobuf==3.20.0

Download datasets

We host the datasets on this link All data gets downloaded and stored in data\impact directory.

How to run

To run each experiment, navigate into HCMT-main. Then, run the following command:

Train a model:

python -m run_model --mode=train

Generate trajectory rollouts:

python -m run_model --mode=eval