16 lines
301 B
Python
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()
|