‹ projects

cluster-rnn

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

init.lua (413B)

      1 ------ parameter server and client ------
      2 require 'mpiT'
      3 mpiT.tag_ps_recv_init  = 1
      4 mpiT.tag_ps_recv_grad  = 2
      5 mpiT.tag_ps_send_param = 3
      6 mpiT.tag_ps_recv_param = 4
      7 mpiT.tag_ps_recv_header = 5
      8 mpiT.tag_ps_recv_stop = 6
      9 mpiT.tag_ps_recv_param_tail = 7
     10 mpiT.tag_ps_recv_grad_tail = 8
     11 
     12 dofile('pserver.lua')
     13 dofile('pclient.lua')
     14 
     15 --dofile('optim-msgd.lua')
     16 --dofile('optim-downpour.lua')
     17 dofile('optim-eamsgd.lua')