turned off visualisation, run dqn cartpole on floydhub

UNSW-CEEM · Feb 3, 2019 · a94cc84 · a94cc84
1 parent d3e8571
commit a94cc84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dqn_cartpole.py b/dqn_cartpole.py
@@ -43,7 +43,7 @@
 # Okay, now it's time to learn something! We visualize the training here for show, but this
 # slows down training quite a lot. You can always safely abort the training prematurely using
 # Ctrl + C.
-dqn.fit(env, nb_steps=500, visualize=False, verbose=2)
+dqn.fit(env, nb_steps=50000, visualize=False, verbose=2)
 
 # After training is done, we save the final weights.
 dqn.save_weights('dqn_{}_weights.h5f'.format(ENV_NAME), overwrite=True)

diff --git a/floyd_example.sh b/floyd_example.sh
@@ -0,0 +1 @@
+floyd run --gpu --env tensorflow-1.3 "python dqn_cartpole.py"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		floyd run --gpu --env tensorflow-1.3 "python dqn_cartpole.py"