@ Override public void finishTraining ( long originatorId , long taskId ) { // on Master thread we'll be applying final gradients if ( params != null && stepFunction != null ) { if ( hasSomething . get ( ) ) { stepFunction . step ( params , updates ) ; //Nd4j.getMemoryManager().memset(updates); updates . assign ( 0.0 ) ; } } }
This method is used on Master only applies buffered updates to params