AdamSolver¶

class dragon.vm.caffe.AdamSolver( solver_file, is_root=True )[source]¶

Examples:

solver {
  base_lr=0.001,
  momentum=0.9,
  momentum2=0.999,
  delta=1e-8,
}

init¶

AdamSolver.__init__( solver_file, is_root=True )[source]¶

Create a AdamSolver.

Parameters:

solver_file (str) – The path of solver file.
is_root (bool, optional, default=True) – True to indicate a root solver.

Solver.base_lr

Return or Set the current learning rate.

Solver.iter

Return or Set the current iteration.

Solver.net

Return the train net.

Solver.test_nets

Return the test nets.

Solver.step(num_iterations=1)[source]

Step the train net.

Parameters:

num_iterations (int, optional, default=1) – The number of iterations to step.