model: Split experiment run()

This splits experiment.run() into 4 different operations: swap_in,
install_prototype, bootstrap_prototype and swap_out.

2 files changed, 10 insertions, 2 deletions

diff --git a/examples/example.py b/examples/example.py
index 54ecd37..8d91102 100755
--- a/examples/example.py
+++ b/examples/example.py
@@ -40,4 +40,8 @@ exp = rl.Experiment(tb, nodes = [a, b])
 
 print(exp)
 
-exp.run()
+try:
+    exp.swap_in()
+    exp.bootstrap_prototype()
+finally:
+    exp.swap_out()
diff --git a/examples/two-layers.py b/examples/two-layers.py
index 9032588..fff3866 100755
--- a/examples/two-layers.py
+++ b/examples/two-layers.py
@@ -52,4 +52,8 @@ exp = rl.Experiment(tb, nodes = [a, b, c, d])
 
 print(exp)
 
-exp.run()
+try:
+    exp.swap_in()
+    exp.bootstrap_prototype()
+finally:
+    exp.swap_out()