-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels_grad1.lua
50 lines (41 loc) · 1.11 KB
/
models_grad1.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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
require 'grad_model'
function create_grad_model5(nc,no)
local grad=create_grad_model(nc,no)
local model=nn.Sequential()
model:add(grad)
return model
end
function create_grad_model1(nc,no)
local grad1=create_grad_model(nc,no)
local model1=nn.Sequential()
model1:add(nn.SpatialMaxPooling(2,2,2,2))
model1:add(grad1)
return model1
end
function create_grad_model2(nc,no)
local grad2=create_grad_model(nc,no)
local model2=nn.Sequential()
model2:add(nn.SpatialMaxPooling(2,2,2,2))
model2:add(nn.SpatialMaxPooling(2,2,2,2))
model2:add(grad2)
return model2
end
function create_grad_model3(nc,no)
local grad3=create_grad_model(nc,no)
local model3=nn.Sequential()
model3:add(nn.SpatialMaxPooling(2,2,2,2))
model3:add(nn.SpatialMaxPooling(2,2,2,2))
model3:add(nn.SpatialMaxPooling(2,2,2,2))
model3:add(grad3)
return model3
end
function create_grad_model4(nc,no)
local grad4=create_grad_model(nc,no)
local model4=nn.Sequential()
model4:add(nn.SpatialMaxPooling(2,2,2,2))
model4:add(nn.SpatialMaxPooling(2,2,2,2))
model4:add(nn.SpatialMaxPooling(2,2,2,2))
model4:add(nn.SpatialMaxPooling(2,2,2,2))
model4:add(grad4)
return model4
end