Emil Wärnberg emiwar
emiwar
/ benchmark_mujoco_rollouts.py
Created
March 18, 2024 20:59
Naive benchmarking (i.e. wall-clock with time.time()) of python-mutlithreaded rollouts of a MuJoCo environment.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import mujoco | |
| import mujoco.rollout | |
| import tqdm | |
| import numpy as np | |
| import pandas as pd | |
| import itertools | |
| import time | |
| model_filename = 'humanoid.xml' | |
| model = mujoco.MjModel.from_xml_path(model_filename) |