Ejemplos de SAC.get_argument en Python

Ejemplo n.º 1

Archivo: sac_ae.py Proyecto: keiohta/tf2rl

    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

Ejemplo n.º 2

Archivo: MFBox.py Proyecto: Chenaah/RAMCO

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)

Ejemplo n.º 3

Archivo: sac_discrete.py Proyecto: ykawamura96/tf2rl

 def get_argument(parser=None):
     parser = SAC.get_argument(parser)
     parser.add_argument('--target-update-interval', type=int, default=None)
     return parser

Ejemplo n.º 4

Archivo: simulation_test.py Proyecto: RobotEmperor/SDU-RL-Manipulator

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()

Ejemplo n.º 5

 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