initial commit
This commit is contained in:
26
main.py
Normal file
26
main.py
Normal file
@ -0,0 +1,26 @@
|
||||
"""
|
||||
Lunar Lander example
|
||||
"""
|
||||
|
||||
import gym
|
||||
|
||||
from model import ActorCritic
|
||||
from run import run
|
||||
from train import train
|
||||
|
||||
seed = 12345
|
||||
|
||||
if __name__ == "__main__":
|
||||
env = gym.make('LunarLander-v2')
|
||||
env.seed(seed)
|
||||
|
||||
policy = ActorCritic()
|
||||
|
||||
training = True
|
||||
|
||||
if training:
|
||||
train(env, policy)
|
||||
|
||||
run(env, policy)
|
||||
|
||||
|
Reference in New Issue
Block a user