Update ACVP

This commit is contained in:
Shangtong Zhang committed 2018-04-10 11:32:59 -06:00
1 parent af1cc11669
commit 669c366bc0
3 files changed
+6 -3

No files matched your search

+3 -3
View File
@@ -302,8 +302,8 @@ def action_conditional_video_prediction():
# a2c_pixel_atari(game)
# Generate a dataset with the trained model
a2c_model_file = './data/A2CAgent-vanilla-model-%s.bin' % (game)
generate_dataset(game, a2c_model_file, prefix)
# a2c_model_file = './data/A2CAgent-vanilla-model-%s.bin' % (game)
# generate_dataset(game, a2c_model_file, prefix)
# Train the action conditional video prediction model
acvp_train(game, prefix)
@@ -334,7 +334,7 @@ if __name__ == '__main__':
# ddpg_continuous()
# ppo_continuous()
action_conditional_video_prediction()
# action_conditional_video_prediction()
# plot()