Skip to content

Commit

Permalink
Fixed GPU device index for gpu and cpu modes
Browse files Browse the repository at this point in the history
  • Loading branch information
kuz committed Aug 15, 2015
1 parent 7e5feff commit 02f5967
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test_gpu
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 02f5967

Please sign in to comment.