Commit 3bbebc3e authored by miaecle's avatar miaecle
Browse files

debug

parent 36e6d64d
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -84,7 +84,8 @@ class MultitaskGraphClassifier(Model):
               pad_batches=True,
               verbose=True):

    super().__init__(self, model_dir=logdir, verbose=verbose)
    super(MultitaskGraphClassifier, self).__init__(
        model_dir=logdir, verbose=verbose)
    self.n_tasks = n_tasks
    self.final_loss = final_loss
    self.model = model
+2 −1
Original line number Diff line number Diff line
@@ -43,7 +43,8 @@ class MultitaskGraphRegressor(Model):
               pad_batches=True,
               verbose=True):

    super().__init__(self, model_dir=logdir, verbose=verbose)
    super(MultitaskGraphRegressor, self).__init__(
        model_dir=logdir, verbose=verbose)
    self.n_tasks = n_tasks
    self.final_loss = final_loss
    self.model = model