ZF/vit-pytorch
1
0
Fork 0
mirror of https://github.com/lucidrains/vit-pytorch.git synced 2025-12-30 08:02:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Phil Wang
2021-11-14 12:24:48 -08:00
parent 99c44cf5f6
commit dc57c75478
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vit_pytorch/mae.py
View File

@@ -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