Setup 1-step prediction

This commit is contained in:
Shangtong Zhang
2017-12-16 22:14:40 -07:00
parent 214f7693c2
commit f0940e86dc
4 changed files with 154 additions and 2 deletions
+4 -1
View File
@@ -2,6 +2,7 @@ import logging
from agent import *
from component import *
from utils import *
import model.action_conditional_video_prediction as acvp
def dqn_cart_pole():
config = Config()
@@ -275,7 +276,7 @@ if __name__ == '__main__':
# dqn_cart_pole()
# async_cart_pole()
a3c_cart_pole()
# a3c_cart_pole()
# a3c_continuous()
# p3o_continuous()
# d3pg_continuous()
@@ -291,3 +292,5 @@ if __name__ == '__main__':
# async_pixel_atari('BreakoutNoFrameskip-v4')
# a3c_pixel_atari('BreakoutNoFrameskip-v4')
acvp.train('PongNoFrameskip-v4')