Update README.md
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ from stable_baselines3.common.env_util import make_vec_env
|
|
| 40 |
gymnasium.register_envs(gymnasium_robotics)
|
| 41 |
|
| 42 |
# Load the trained model
|
| 43 |
-
model = TQC.load(
|
| 44 |
|
| 45 |
# Create the environment
|
| 46 |
env = make_vec_env("FetchPushDense-v4", n_envs=1)
|
|
|
|
| 40 |
gymnasium.register_envs(gymnasium_robotics)
|
| 41 |
|
| 42 |
# Load the trained model
|
| 43 |
+
model = TQC.load("best-model.zip")
|
| 44 |
|
| 45 |
# Create the environment
|
| 46 |
env = make_vec_env("FetchPushDense-v4", n_envs=1)
|