Search code examples
gpurestartconvolutioncaffe

In CNN with caffe, Can I set up initial caffemodel?

I have operated CNN using caffe. however, system was forced termination. I have caffemodel so far.

Can I restart learning from now on using current caffemodel?

Thanks,

Solution

  • Caffe supports resuming as explained here:

    We all experience times when the power goes out [...] Since we are snapshotting intermediate results during training, we will be able to resume from snapshots.

    This is available via the --snapshot option of the main caffe command-line tool, e.g:

    ./build/tools/caffe train [...] --snapshot=caffenet_train_10000.solverstate

    As explained within the doc caffenet_train_10000.solverstate is:

    the solver state snapshot that stores all necessary information to recover the exact solver state.

    In particular you can find more precisions about how to configure snapshotting from the solver documentation.