Upgrade black to 22.1.0 and fix the corresponding issues

2022-03-20 23:18:23 -07:00
parent 8d0799dfb1
commit 676e8e411d
3 changed files with 3 additions and 3 deletions
--- a/xautodl/procedures/optimizers.py
+++ b/xautodl/procedures/optimizers.py
@@ -155,7 +155,7 @@ class ExponentialLR(_LRScheduler):
            if self.current_epoch >= self.warmup_epochs:
                last_epoch = self.current_epoch - self.warmup_epochs
                assert last_epoch >= 0, "invalid last_epoch : {:}".format(last_epoch)
-                lr = base_lr * (self.gamma ** last_epoch)
+                lr = base_lr * (self.gamma**last_epoch)
            else:
                lr = (
                    self.current_epoch / self.warmup_epochs