1 files changed, 5 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index 34bb793..287d0c2 100644
--- a/init.lua
+++ b/init.lua
@@ -3,8 +3,13 @@ require 'torch'
 
 optim = {}
 
+-- optimizations
 torch.include('optim', 'sgd.lua')
 torch.include('optim', 'cg.lua')
 torch.include('optim', 'asgd.lua')
 torch.include('optim', 'fista.lua')
 torch.include('optim', 'lbfgs.lua')
+
+-- tools 
+torch.include('optim', 'ConfusionMatrix.lua')
+torch.include('optim', 'Logger.lua')