-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.m
45 lines (43 loc) · 1.38 KB
/
config.m
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
%% network setting
global trainOpts;
trainOpts.FilterNumber = 8;
trainOpts.FilterSize = 3;
trainOpts.Stage = 7;
trainOpts.Initer = 3; % inversion block number
trainOpts.Padding = 1;
trainOpts.LinearLabel = double(-1:0.02:1); % fixed
%% training and testing setting
trainOpts.sigma = 50;
trainOpts.Fold = 1;
trainOpts.batchSize = 1 ;
trainOpts.learningRate = 0.1;
trainOpts.numSubBatches = 1 ;
trainOpts.epochSize = inf;
trainOpts.prefetch = false ;
trainOpts.numEpochs = 300 ; % training number
trainOpts.weightDecay = 0.0005 ;
trainOpts.momentum = 0.95 ;
trainOpts.randomSeed = 0 ;
trainOpts.sync = false ;
trainOpts.cudnn = true ;
trainOpts.backPropDepth = +inf ;
trainOpts.continue = false ;
trainOpts.gpus = [] ;;
trainOpts.expDir = fullfile('Train_output','net') ;
trainOpts.train = [] ;
trainOpts.val = [] ;
trainOpts.solver = [] ; % Empty array means use the default SGD solver
trainOpts.saveSolverState = true ;
trainOpts.nesterovUpdate = false ;
trainOpts.memoryMapFile = fullfile(tempdir, 'matconvnet.bin') ;
trainOpts.profile = false ;
trainOpts.parameterServer.method = 'mmap' ;
trainOpts.parameterServer.prefix = 'mcn' ;
trainOpts.conserveMemory = false ;
trainOpts.backPropDepth = +inf ;
trainOpts.errorFunction = 'none';%multiclass' ;
trainOpts.errorLabels = {} ;
trainOpts.plotDiagnostics = false ;
trainOpts.plotStatistics = true;
trainOpts.postEpochFn = [] ;
trainOpts.skipForward = false ;