def main():
    args = energyplus_arg_parser().parse_args()
    train(args.env, num_timesteps=args.num_timesteps, seed=args.seed)
Exemplo n.º 2
0
from baselines_energyplus.trpo_mpi.run_energyplus import train
from baselines_energyplus.common.energyplus_util import energyplus_arg_parser
import sys
'''PLEASE KEEP THIS LINE'''
from algorithms import *

# using customized learning algorithms and policy functions

if __name__ == '__main__':
    args = energyplus_arg_parser().parse_args()

    train(env_id=args.env,
          num_timesteps=args.num_timesteps,
          seed=args.seed,
          learn=getattr(sys.modules['algorithms'], args.learn),
          policy_fn_class=eval(args.policy))