hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update MuJoCo examples to use Ant-v4 instead of Ant-v3

Browse Source
This commit is contained in:
Dominik Jain 2024-01-10 15:39:53 +01:00
parent eaab7b0a4b
commit 7fa588309b
13 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mujoco/mujoco_a2c.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=4096) parser.add_argument("--buffer-size", type=int, default=4096)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64])

2
examples/mujoco/mujoco_ddpg.py
View File

@ -21,7 +21,7 @@ from tianshou.utils.net.continuous import Actor, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=1000000) parser.add_argument("--buffer-size", type=int, default=1000000)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256])

2
examples/mujoco/mujoco_ddpg_hl.py
View File

@ -17,7 +17,7 @@ from tianshou.utils.logging import datetime_tag
def main( def main(
experiment_config: ExperimentConfig, experiment_config: ExperimentConfig,
task: str = "Ant-v3", task: str = "Ant-v4",
buffer_size: int = 1000000, buffer_size: int = 1000000,
hidden_sizes: Sequence[int] = (256, 256), hidden_sizes: Sequence[int] = (256, 256),
actor_lr: float = 1e-3, actor_lr: float = 1e-3,

2
examples/mujoco/mujoco_npg.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=4096) parser.add_argument("--buffer-size", type=int, default=4096)
parser.add_argument( parser.add_argument(

2
examples/mujoco/mujoco_npg_hl.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.logging import datetime_tag
def main( def main(
experiment_config: ExperimentConfig, experiment_config: ExperimentConfig,
task: str = "Ant-v3", task: str = "Ant-v4",
buffer_size: int = 4096, buffer_size: int = 4096,
hidden_sizes: Sequence[int] = (64, 64), hidden_sizes: Sequence[int] = (64, 64),
lr: float = 1e-3, lr: float = 1e-3,

2
examples/mujoco/mujoco_ppo.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=4096) parser.add_argument("--buffer-size", type=int, default=4096)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64])

2
examples/mujoco/mujoco_redq.py
View File

@ -20,7 +20,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=1000000) parser.add_argument("--buffer-size", type=int, default=1000000)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256])

2
examples/mujoco/mujoco_reinforce.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.net.continuous import ActorProb
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=4096) parser.add_argument("--buffer-size", type=int, default=4096)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[64, 64])

2
examples/mujoco/mujoco_reinforce_hl.py
View File

@ -20,7 +20,7 @@ from tianshou.utils.logging import datetime_tag
def main( def main(
experiment_config: ExperimentConfig, experiment_config: ExperimentConfig,
task: str = "Ant-v3", task: str = "Ant-v4",
buffer_size: int = 4096, buffer_size: int = 4096,
hidden_sizes: Sequence[int] = (64, 64), hidden_sizes: Sequence[int] = (64, 64),
lr: float = 1e-3, lr: float = 1e-3,

2
examples/mujoco/mujoco_sac.py
View File

@ -20,7 +20,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=1000000) parser.add_argument("--buffer-size", type=int, default=1000000)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256])

2
examples/mujoco/mujoco_td3.py
View File

@ -21,7 +21,7 @@ from tianshou.utils.net.continuous import Actor, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=1000000) parser.add_argument("--buffer-size", type=int, default=1000000)
parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256]) parser.add_argument("--hidden-sizes", type=int, nargs="*", default=[256, 256])

2
examples/mujoco/mujoco_trpo.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.net.continuous import ActorProb, Critic
def get_args(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--task", type=str, default="Ant-v3") parser.add_argument("--task", type=str, default="Ant-v4")
parser.add_argument("--seed", type=int, default=0) parser.add_argument("--seed", type=int, default=0)
parser.add_argument("--buffer-size", type=int, default=4096) parser.add_argument("--buffer-size", type=int, default=4096)
parser.add_argument( parser.add_argument(

2
examples/mujoco/mujoco_trpo_hl.py
View File

@ -23,7 +23,7 @@ from tianshou.utils.logging import datetime_tag
def main( def main(
experiment_config: ExperimentConfig, experiment_config: ExperimentConfig,
task: str = "Ant-v3", task: str = "Ant-v4",
buffer_size: int = 4096, buffer_size: int = 4096,
hidden_sizes: Sequence[int] = (64, 64), hidden_sizes: Sequence[int] = (64, 64),
lr: float = 1e-3, lr: float = 1e-3,