docs: @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);
              }
          }
  
      }