imports

Lightning-Universe · Oct 19, 2020 · 097fd22 · 097fd22
1 parent 0879a96
commit 097fd22
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_bolts/callbacks/self_supervised.py b/pl_bolts/callbacks/self_supervised.py
@@ -103,7 +103,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
  pl_module.logger.log_metrics(metrics, step=trainer.global_step)
 
 
-class BYOLMAWeightUpdate(pl.Callback):
+class BYOLMAWeightUpdate(Callback):
  """
  Weight update rule from BYOL.