docs: public synchronized void fit(INDArray features, INDArray labels, INDArray featuresMask, INDArray labelsMask) {
          try{
              fitHelper(features, labels, featuresMask, labelsMask);
          } catch (OutOfMemoryError e){
              CrashReportingUtil.writeMemoryCrashDump(this, e);
              throw e;
          }
      }