Permalink
Browse files

Fixed GPU device index for gpu and cpu modes

  • Loading branch information...
1 parent 7e5feff commit 02f5967c6dcdd5528da715dcac37385baec3843d @kuz committed Aug 15, 2015
Showing with 1 addition and 1 deletion.
  1. +1 −1 test_gpu
View
@@ -36,7 +36,7 @@ state_dim=7056
ncols=1
agent_params="lr="$lr",ep=1,ep_end="$eps_end",ep_endt="$eps_endt",discount="$discount",hist_len=4,learn_start="$learn_start",replay_memory="$replay_memory",update_freq="$update_freq",n_replay="$n_replay",network="$netfile",preproc="$preproc_net",state_dim="$state_dim",minibatch_size=32,rescale_r=1,ncols="$ncols",bufferSize=512,valid_size=500,target_q=10000,clip_delta=1,min_reward=-1,max_reward=1"
gif_file="../gifs/$ENV.gif"
-gpu=1
+gpu=0
random_starts=30
pool_frms="type="$pool_frms_type",size="$pool_frms_size
num_threads=4

0 comments on commit 02f5967

Please sign in to comment.