Misc

Quick Reference

List Brief
caffe.set_mode_cpu Set to the CPU mode.
caffe.set_mode_gpu Set to the GPU mode.
caffe.set_device Set the active device.
caffe.set_random_seed Set the global random seed.
caffe.root_solver Whether this node is root.
caffe.set_root_solver Set this node to the root.

API Reference

Implementation the singleton utility.

dragon.vm.caffe.misc.set_mode_cpu()

Set to the CPU mode. [PyCaffe Style]

Returns:
Return type:None

References

The implementation of set_mode_cpu(_caffe.cpp, L51).

dragon.vm.caffe.misc.set_mode_gpu()

Set to the GPU mode. [PyCaffe Style]

Returns:
Return type:None

References

The implementation of set_mode_gpu(_caffe.cpp, L52).

dragon.vm.caffe.misc.set_device(device)

Set the active device. [PyCaffe Style]

Returns:
Return type:None

References

The implementation of SetDevice(common.cpp, L65).

dragon.vm.caffe.misc.set_random_seed(seed)

Set the global random seed. [PyCaffe Style]

Parameters:seed (int) – The random seed.
Returns:
Return type:None

References

The implementation of set_random_seed(_caffe.cpp, L71).

dragon.vm.caffe.misc.root_solver()

Whether this node is root.

Returns:True, if setting it before.
Return type:boolean

References

The implementation of root_solver(common.hpp, L164).

dragon.vm.caffe.misc.set_root_solver(val)

Set this node to the root.

Parameters:val (boolean) – Whether to become the root.

References

The implementation of set_root_solver(common.hpp, L165).