Skip to content

Commit

Permalink
less nThreads
Browse files Browse the repository at this point in the history
  • Loading branch information
bearpaw committed May 8, 2017
1 parent bf5f28a commit a4d14d6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion train/exp/train.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
cd ..

# Example for training with 4x 12GB GPUs
# You may set larger -nThreads for faster data loading
# Please refer to opts.lua for more training options
CUDA_VISIBLE_DEVICES=0,1,2,3 th main.lua -netType hg-attention -expID crf_parts -nGPU 1 -LRNKer 11 -nGPU 4
CUDA_VISIBLE_DEVICES=0,1,2,3 th main.lua -netType hg-attention -expID crf_parts -nGPU 1 -LRNKer 11 -nGPU 4 -nThreads 2

0 comments on commit a4d14d6

Please sign in to comment.