def get_argument(parser=None):
"""
Create or update argument parser for command line program
Args:
parser (argparse.ArgParser, optional): argument parser
Returns:
argparse.ArgParser: argument parser
"""
parser = SAC.get_argument(parser)
parser.add_argument('--stop-q-grad', action="store_true")
parser.add_argument('--memory-capacity', type=int, default=int(1e5))
return parser
def main():
parser = Trainer.get_argument()
parser = OnPolicyTrainer.get_argument(parser)
parser = SAC.get_argument(parser)
parser = PPO.get_argument(parser)
parser = TD3.get_argument(parser)
parser.add_argument('--SAC', action='store_true')
parser.add_argument('--PPO', action='store_true')
parser.add_argument('--TD3', action='store_true')
parser.add_argument('--DEBUG', action='store_true')
parser.add_argument('--env', type=int, default=0)
parser.set_defaults(batch_size=32) #100
parser.set_defaults(n_warmup=10000) #10000
parser.set_defaults(max_steps=2e6)
parser.set_defaults(gpu=0)
parser.set_defaults(test_interval=200 * 100)
parser.set_defaults(test_episodes=3)
args = parser.parse_args()
print(vars(args))
run(parser)
def get_argument(parser=None):
parser = SAC.get_argument(parser)
parser.add_argument('--target-update-interval', type=int, default=None)
return parser
import numpy as np
import gym
from tf2rl.algos.sac import SAC
from tf2rl.experiments.trainer import Trainer
import tensorflow as tf
parser = Trainer.get_argument()
parser = SAC.get_argument(parser)
# parser.add_argument('--env-name', type=str, default="CartPole-v0")
parser.add_argument('--env-name', type=str, default="belt_task:belt-task-v0")
parser.add_argument('--seed', type=int, default=42, help='random seed')
parser.add_argument('--render',
type=bool,
default=False,
help='set gym environment to render display')
parser.add_argument('--verbose',
type=bool,
default=False,
help='log execution details')
parser.add_argument(
'--model_path',
type=str,
default=
'/home/yik/catkin_ws/src/SDU-RL-Manipulator/rl_motion_trj/scripts/results/',
help='path to save model')
parser.add_argument('--model_name',
type=str,
default=f'',
help='name of the saved model')
args = parser.parse_args()
def get_argument(parser=None):
parser = SAC.get_argument(parser)
parser.add_argument('--stop-q-grad', action="store_true")
parser.add_argument('--memory-capacity', type=int, default=int(1e5))
return parser