/** * Expects a computation graph where some vertices are frozen * * @param orig */ public TransferLearningHelper(ComputationGraph orig) { origGraph = orig; initHelperGraph(); }
/** * Will modify the given comp graph (in place!) to freeze vertices from input to the vertex specified. * * @param orig Comp graph * @param frozenOutputAt vertex to freeze at (hold params constant during training) */ public TransferLearningHelper(ComputationGraph orig, String... frozenOutputAt) { origGraph = orig; this.frozenOutputAt = frozenOutputAt; applyFrozen = true; initHelperGraph(); }