No description, website, or topics provided.
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
CMakeLists.txt
README.md
download_mnist.py
mnist.cpp

README.md

libtorch-mnist

cmake -DCUDA_HOST_COMPILER=/usr/bin/gcc-5 -DCMAKE_PREFIX_PATH=/path/to/libtorch ..

You need to call the fuction as following:

  • train(epoch, options, model, device, *train_loader, optimizer, dataset_size.value());
  • test(model, device, *test_loader, dataset_size.value());