Skip to content

Commit

Permalink
correct small details
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosuc3m committed Dec 12, 2023
1 parent 87acc08 commit dd15bf5
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -368,7 +368,7 @@ void setFineTuningData(List<Tensor<T>> trainingSamples, List<Tensor<T>> groundTr
* single tensor that corresponds to the groudn truth to fine tune the model
*/
public < T extends RealType< T > & NativeType< T > >
void setFineTuningData(Tensor<T> trainingSamples, Tensor<T> groundTruth) throws IOException, Exception {
void setFineTuningData(Tensor<T> trainingSamples, Tensor<T> groundTruth) {
checkTrainAndGroundTruthDimensions(trainingSamples, groundTruth);
setTrainingSamples(trainingSamples);
setGroundTruth(groundTruth);
Expand All @@ -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;
Expand Down

0 comments on commit dd15bf5

Please sign in to comment.