cleanup sepvit

This commit is contained in:
Phil Wang
2022-03-31 14:35:11 -07:00
parent d65a742efe
commit 5d4c798949
2 changed files with 1 additions and 3 deletions

View File

@@ -253,8 +253,6 @@ class SepViT(nn.Module):
dropout = 0.
):
super().__init__()
self.to_patches = nn.Conv2d(channels, dim, 7, stride = 4, padding = 3)
assert isinstance(depth, tuple), 'depth needs to be tuple if integers indicating number of transformer blocks at that stage'
num_stages = len(depth)