(self)
| 10 | |
| 11 | class TestSolver(unittest.TestCase): |
| 12 | def setUp(self): |
| 13 | self.num_output = 13 |
| 14 | net_f = simple_net_file(self.num_output) |
| 15 | f = tempfile.NamedTemporaryFile(mode='w+', delete=False) |
| 16 | f.write("""net: '""" + net_f + """' |
| 17 | test_iter: 10 test_interval: 10 base_lr: 0.01 momentum: 0.9 |
| 18 | weight_decay: 0.0005 lr_policy: 'inv' gamma: 0.0001 power: 0.75 |
| 19 | display: 100 max_iter: 100 snapshot_after_train: false |
| 20 | snapshot_prefix: "model" """) |
| 21 | f.close() |
| 22 | self.solver = caffe.SGDSolver(f.name) |
| 23 | # also make sure get_solver runs |
| 24 | caffe.get_solver(f.name) |
| 25 | caffe.set_mode_cpu() |
| 26 | # fill in valid labels |
| 27 | self.solver.net.blobs['label'].data[...] = \ |
| 28 | np.random.randint(self.num_output, |
| 29 | size=self.solver.net.blobs['label'].data.shape) |
| 30 | self.solver.test_nets[0].blobs['label'].data[...] = \ |
| 31 | np.random.randint(self.num_output, |
| 32 | size=self.solver.test_nets[0].blobs['label'].data.shape) |
| 33 | os.remove(f.name) |
| 34 | os.remove(net_f) |
| 35 | |
| 36 | def test_solve(self): |
| 37 | self.assertEqual(self.solver.iter, 0) |
nothing calls this directly
no test coverage detected