diff options
Diffstat (limited to 'test/test_graph.lua')
-rw-r--r-- | test/test_graph.lua | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/test/test_graph.lua b/test/test_graph.lua index 7f33581..34857b3 100644 --- a/test/test_graph.lua +++ b/test/test_graph.lua @@ -1,9 +1,7 @@ - require 'graph' -require 'totem' - -local tester = totem.Tester() -local tests = {} +require 'torch' +local tester = torch.Tester() +local tests = torch.TestSuite() local function create_graph(nlayers, ninputs, noutputs, nhiddens, droprate) local g = graph.Graph() |