try sep vit

2024-04-28 01:01:31 +07:00 · 2024-04-28 01:01:31 +07:00 · a246d2bb64
commit a246d2bb64
parent 4a962a02ad
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -1,5 +1,5 @@
 import torch
-from torch import nn
+from torch import nn, einsum
 import torch.nn.functional as F
 import numpy as np
 from functools import partial