benoitdescamps

class Reptile:
  """
  Repile-optimization as described by Ravi,et.al. (https://openreview.net/pdf?id=rJY0-Kcl)
  """
  def __init__(self,
               model:torch.nn.Module,
               metalearners:List[MetaLearner]):
    self.n_tasks = len(metalearners)
    self.model = model
    self.metalearners = metalearners

benoitdescamps

class MetaLearner:
  """
  This is nothing more than a regular learning flow. However, we create this 
  class, as we plan on using separate (meta-)learners for each task.
  """

  def __init__(self,
               model:torch.nn.Module,
               loss_fn:Callable,
               optimizer):