cleanup

smilekitty7 · Nov 14, 2021 · dc57c75 · dc57c75
1 parent 99c44cf
commit dc57c75
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/vit_pytorch/mae.py b/vit_pytorch/mae.py
@@ -1,8 +1,7 @@
 import torch
-from math import ceil
 from torch import nn
 import torch.nn.functional as F
-from einops import rearrange, repeat
+from einops import repeat
 
 from vit_pytorch.vit import Transformer