Skip to content
This repository has been archived by the owner on Jul 24, 2024. It is now read-only.

Commit

Permalink
Update model.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
lepton-flux authored Oct 16, 2023
1 parent 32538d6 commit 295f69c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/model.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ int main() {
std::cout << "Epoch: " << epoch + 1 << std::endl;

for(int batch = 0; batch < dataset.lenght(); ++batch) {
input.copy(dataset.features()[batch].internal()); // this could be done without copies with a view.
input.copy(dataset.features()[batch].internal()); // I will fix this in the futuro so it will be prettier and without copies.
targets.copy(dataset.targets()[batch].internal());

std::cout << "loss" << criterion.loss() << std::endl;
Expand All @@ -51,4 +51,4 @@ int main() {
}

}
}
}

0 comments on commit 295f69c

Please sign in to comment.