add original

JothiRamaswamy · Oct 20, 2022 · e7bf583 · e7bf583
1 parent 31711af
commit e7bf583
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/unlearn/__init__.py b/unlearn/__init__.py
@@ -4,11 +4,16 @@
 from .fisher import fisher
 from .retrain import retrain
 
+def raw(data_loaders, model, criterion, args):
+    pass
+
 def get_unlearn_method(name):
     """ method usage:
     
     function(data_loaders, model, criterion, args)"""
-    if name == "RL":
+    if name == "raw":
+        return raw
+    elif name == "RL":
         return RL
     elif name == "GA":
         return GA