Add top baseline

2021-07-07 17:46:11 +02:00
parent 6658b8c7f0
commit 5855f67c2a
4 changed files with 68 additions and 40 deletions
--- a/active_grasp/init.py
+++ b/active_grasp/init.py
@@ -0,0 +1,5 @@
+from .policy import register
+from .baselines import *
+
+register("single-view", SingleViewBaseline)
+register("top", TopBaseline)