caffe

1
classdef test_solver < matlab.unittest.TestCase
2
  
3
  properties
4
    num_output
5
    solver
6
  end
7
  
8
  methods
9
    function self = test_solver()
10
      self.num_output = 13;
11
      model_file = caffe.test.test_net.simple_net_file(self.num_output);
12
      solver_file = tempname();
13
      
14
      fid = fopen(solver_file, 'w');
15
      fprintf(fid, [ ...
16
        'net: "'  model_file  '"\n' ...
17
        'test_iter: 10 test_interval: 10 base_lr: 0.01 momentum: 0.9\n' ...
18
        'weight_decay: 0.0005 lr_policy: "inv" gamma: 0.0001 power: 0.75\n' ...
19
        'display: 100 max_iter: 100 snapshot_after_train: false\n' ]);
20
      fclose(fid);
21
      
22
      self.solver = caffe.Solver(solver_file);
23
      % also make sure get_solver runs
24
      caffe.get_solver(solver_file);
25
      caffe.set_mode_cpu();
26
      % fill in valid labels
27
      self.solver.net.blobs('label').set_data(randi( ...
28
        self.num_output - 1, self.solver.net.blobs('label').shape));
29
      self.solver.test_nets(1).blobs('label').set_data(randi( ...
30
        self.num_output - 1, self.solver.test_nets(1).blobs('label').shape));
31
      
32
      delete(solver_file);
33
      delete(model_file);
34
    end
35
  end
36
  methods (Test)
37
    function test_solve(self)
38
      self.verifyEqual(self.solver.iter(), 0)
39
      self.solver.step(30);
40
      self.verifyEqual(self.solver.iter(), 30)
41
      self.solver.solve()
42
      self.verifyEqual(self.solver.iter(), 100)
43
    end
44
  end
45
end
46