/diffsim

Differentiable simulator

Primary LanguagePython

Differentiable Simulation

Differentiable robot-terrain interaction model for a tracked robot written using NVIDIA-Warp.

Running

To train Terrain Encoder model with the L2-loss computed between predicted (by Differentiable Physics) and ground truth trajectories, run:

python terrain_optimization

Weights

The pretrained weights for the LSS Terrain Encoder can be downloaded from:

Once downloaded, please put the weights to config/weights/ folder.