tree: 1742f78ec8a7af2b6db431053ee97cc89cb884bc [path history] [tgz]
  1. CMakeLists.txt
  2. README.md
  3. time_observer.cc
  4. time_observer.h
  5. time_observer_test.cc
caffe2/observers/README.md

Observers

Usage

Observers are a small framework that allow users to attach code to the execution of SimpleNets and Operators.

An example of an Observer is the TimeObserver, used as follows:

C++

unique_ptr<TimeObserver<NetBase>> net_ob =
    make_unique<TimeObserver<NetBase>>(net.get());
auto* ob = net->AttachObserver(std::move(net_ob));
net->Run();
LOG(INFO) << "av time children: " << ob->average_time_children();
LOG(INFO) << "av time: " << ob->average_time();

Python

model.net.AttachTimeObserver()
ws.RunNet(model.net)
ob = model.net.GetObserver()

print("av time children:", ob.average_time_children())
print("av time:", ob.average_time())

Implementing An Observer

To implement an observer you must inherit from ObserverBase and implement the Start and Stop functions.

Observers are instantiated with a subject of a generic type, such as a Net or Operator. The observer framework is built to be generic enough to “observe” various other types, however.