Update the readme and fix bugs in custom-dataset example #1214
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
random_shuffle
: is not a member of 'std' since it c++17, updated toshuffle
.Functional(torch::log_softmax, 1, torch::nullopt)
no longer matches the constructor, updated to lambda expression.auto tlabel = torch::from_blob(&data[index].second, {1}, torch::kLong);
sometimes leads to non-sensical numerical value.A few examples from std::cout:
Updated to direct the creation of a tensor instead of creation from a pointer.
Tested the program via libtorch (2.1.2) with cuda 12.1 using both the release and debug build.