We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26c3048 commit a53bea7Copy full SHA for a53bea7
tests/accuracy/test_int8_accuracy.cpp
@@ -1,6 +1,7 @@
1
#include "accuracy_test.h"
2
#include "datasets/cifar10.h"
3
#include "torch/torch.h"
4
+#include "trtorch/ptq.h"
5
6
TEST_P(AccuracyTests, FP16AccuracyIsClose) {
7
auto calibration_dataset = datasets::CIFAR10("tests/accuracy/datasets/data/cifar-10-batches-bin/", datasets::CIFAR10::Mode::kTest)
0 commit comments