def main(config, modelParam):
if config['network'] == 'CartPole_v1_image':
env = EnvironmentWrapper_image(modelParam)
else:
env = EnvironmentWrapper(modelParam)
# create an instance of the model you want
model = Model(config, modelParam, env)
# create an instacne of the saver and resoterer class
saveRestorer = SaverRestorer(config, modelParam)
model = saveRestorer.restore(model)
# here you train your model
if modelParam['play'] == False:
trainer = Trainer(model, modelParam, config, saveRestorer, env)
trainer.train()
#play
if modelParam['play'] == True:
player = Player(model, modelParam, config, saveRestorer, env)
player.play_episode()
return
