‹ projects

cluster-rnn

a distributed Torch7 RNN cluster over MPI
Log | Files | Refs | README

glaunch.lua (318B)

      1 -- gpu launch
      2 -- Author: Sixin Zhang (zsx@cims.nyu.edu)
      3 -- luajit glaunch.lua
      4 
      5 dofile('init.lua')
      6 
      7 opt = {}
      8 opt.gpuid = 1
      9 
     10 require 'cunn'
     11 torch.setdefaulttensortype('torch.CudaTensor')
     12 print('[client] use gpu', opt.gpuid)
     13 cutorch.setDevice(opt.gpuid)
     14 
     15 opt.name = 'sgd'
     16 opt.lr = 1e-1
     17 opt.mom = 0.99
     18 
     19 dofile('goot.lua')