2025-07-28 12:35:44 +08:00

16 lines
301 B
Python

import pickle
import argparse
parser = argparse.ArgumentParser()
parser.add_argument(
"--latent", type=str, required=False, default="mujoco_saved_latent.pkl"
)
args = parser.parse_args()
latent = pickle.load(open(args.latent, "rb"))
import matplotlib.pyplot as plt
plt.plot(latent)
plt.show()