forked from koraykv/kex
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathinit.lua
More file actions
26 lines (24 loc) · 858 Bytes
/
init.lua
File metadata and controls
26 lines (24 loc) · 858 Bytes
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