-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathinit.lua
26 lines (24 loc) · 858 Bytes
/
init.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
require 'torch'
require 'nn'
require 'libkex'
torch.include('kex','lushio.lua')
-- extra modules that we need
torch.include('kex', 'SpatialFullConvolution.lua')
torch.include('kex', 'SpatialFullConvolutionMap.lua')
torch.include('kex', 'WeightedMSECriterion.lua')
torch.include('kex', 'L1Cost.lua')
torch.include('kex', 'Diag.lua')
torch.include('kex', 'CriterionModule.lua')
torch.include('kex', 'SpatialMaxPyramid.lua')
torch.include('kex', 'SpatialMaxPooling2.lua')
torch.include('kex', 'Vectorize.lua')
torch.include('kex', 'TanhShrink.lua')
torch.include('kex', 'Crop.lua')
torch.include('kex', 'SqrtBias.lua')
torch.include('kex', 'stochasticrates.lua')
torch.include('kex', 'TensorLinear.lua')
torch.include('kex', 'DivisiveNormalization.lua')
torch.include('kex', 'hessian.lua')
function kex.cudahacks()
torch.include('kex','cudahacks.lua')
end