19 lines
412 B
Python
19 lines
412 B
Python
|
from runner import Runner
|
||
|
from common.arguments import get_args
|
||
|
from common.utils import make_env
|
||
|
import numpy as np
|
||
|
import random
|
||
|
import torch
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
# get the params
|
||
|
args = get_args()
|
||
|
env, args = make_env(args)
|
||
|
runner = Runner(args, env)
|
||
|
if args.evaluate:
|
||
|
returns = runner.evaluate()
|
||
|
print('Average returns is', returns)
|
||
|
else:
|
||
|
runner.run()
|