From dd15bf5a1f05d17b036f500a724871dc1fa9089f Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Wed, 13 Dec 2023 00:35:24 +0100 Subject: [PATCH] correct small details --- .../modelrunner/runmode/ops/StardistFineTuneJdllOp.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/main/java/io/bioimage/modelrunner/runmode/ops/StardistFineTuneJdllOp.java b/src/main/java/io/bioimage/modelrunner/runmode/ops/StardistFineTuneJdllOp.java index 244f9419..388640e2 100644 --- a/src/main/java/io/bioimage/modelrunner/runmode/ops/StardistFineTuneJdllOp.java +++ b/src/main/java/io/bioimage/modelrunner/runmode/ops/StardistFineTuneJdllOp.java @@ -368,7 +368,7 @@ void setFineTuningData(List> trainingSamples, List> groundTr * single tensor that corresponds to the groudn truth to fine tune the model */ public < T extends RealType< T > & NativeType< T > > - void setFineTuningData(Tensor trainingSamples, Tensor groundTruth) throws IOException, Exception { + void setFineTuningData(Tensor trainingSamples, Tensor groundTruth) { checkTrainAndGroundTruthDimensions(trainingSamples, groundTruth); setTrainingSamples(trainingSamples); setGroundTruth(groundTruth); @@ -384,9 +384,9 @@ public void setBatchSize(int batchSize) { } /** - * Learning rate used to fine tune the model, if not set, the default one will be used {@link #learningRate} + * Learning rate used to fine tune the model, if not set, the default one will be used {@link #lr} * @param learningRate - * Learning rate used to fine tune the model, if not set, the default one will be used {@link #learningRate} + * Learning rate used to fine tune the model, if not set, the default one will be used {@link #lr} */ public void setLearingRate(float learningRate) { this.lr = learningRate;