SGDSolver¶
- class
dragon.vm.caffe.
SGDSolver
(
solver_file,
is_root=True
)[source]¶ The Momentum-SGD solver. [Polyak, 1964].
Examples:
solver { base_lr=0.01, momentum=0.9, }
__init__¶
Properties¶
base_lr¶
Solver.
base_lr
Return or Set the current learning rate.
- Returns:
- float – The current learning rate.
iter¶
Solver.
iter
Return or Set the current iteration.
- Returns:
- int – The current iteration.
net¶
Solver.
net
Return the train net.
- Returns:
- dragon.vm.caffe.Net – The train net.
test_nets¶
Solver.
test_nets
Return the test nets.
- Returns:
- Sequence[dragon.vm.caffe.Net] – The test nets.