-
Notifications
You must be signed in to change notification settings - Fork 69
/
config.lua
44 lines (37 loc) · 950 Bytes
/
config.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
--mode:
--train
--test
--db:
--ICVL
--NYU
--MSRA
--HANDS2017
--side
--top
--side: ITOP side-view
--top: ITOP top-view
mode = "train"
db = "ICVL"
resume = false
resume_epoch = 0
model_name = "model.net"
epochLimit = 10
nThread = 6
nGPU = 1
gpu_table = {1,2}
defaultGPU = 1
inputDim = 1
worldDim = 3
bkgValue = 0
originalSz = 96
croppedSz = 88
poolFactor = 2
std = 1.7
lr = 2.5e-4
aph = 0.99
eps = 1e-8
batchSz = 8
--predmade grid tensor
d3Output_x = torch.view(torch.range(1,croppedSz/poolFactor),1,1,croppedSz/poolFactor):repeatTensor(croppedSz/poolFactor,croppedSz/poolFactor,1):type('torch.CudaTensor')
d3Output_y = torch.view(torch.range(1,croppedSz/poolFactor),1,croppedSz/poolFactor,1):repeatTensor(croppedSz/poolFactor,1,croppedSz/poolFactor):type('torch.CudaTensor')
d3Output_z = torch.view(torch.range(1,croppedSz/poolFactor),croppedSz/poolFactor,1,1):repeatTensor(1,croppedSz/poolFactor,croppedSz/poolFactor):type('torch.CudaTensor')