‹ projects

cluster-rnn

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

codeblock.lua (626B)

      1 -------------------------------------------------------------------------------
      2 -- Author: umhau (umhau@alum.gcc.edu)
      3 -------------------------------------------------------------------------------
      4 
      5 
      6 
      7 -- NOTES ----------------------------------------------------------------------
      8 
      9 
     10 
     11 -- MPI SETTINGS ---------------------------------------------------------------
     12 -- most of these are set in the mlaunch file.  These are mostly duplicates.
     13 
     14 local state = state or {}
     15 local mb = opt.mb or 128
     16 
     17 require 'optim'
     18 
     19 opti = optim.eamsgd
     20 state.optim = {
     21     lr = lr,
     22     pclient = pclient,
     23     su = su,
     24     mva = mva,
     25     mom = mom,
     26 }