@Override public BaseLayer clone() { BaseLayer clone = (BaseLayer) super.clone(); if (clone.dist != null) clone.dist = clone.dist.clone(); if (clone.learningRateSchedule != null) clone.learningRateSchedule = new HashMap<>(clone.learningRateSchedule); if (clone.momentumSchedule != null) clone.momentumSchedule = new HashMap<>(clone.momentumSchedule); return clone; }