1 files changed, 1 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index 27dfbfd..2d62899 100644
--- a/init.lua
+++ b/init.lua
@@ -47,6 +47,7 @@ torch.include('nnx', 'test-omp.lua')
 -- tools:
 torch.include('nnx', 'ConfusionMatrix.lua')
 torch.include('nnx', 'Logger.lua')
+torch.include('nnx', 'Probe.lua')
 
 -- OpenMP module:
 torch.include('nnx', 'OmpModule.lua')