try sep vit

This commit is contained in:
thanhvc3 2024-04-28 01:01:31 +07:00
parent 4a962a02ad
commit a246d2bb64

View File

@ -1,5 +1,5 @@
import torch import torch
from torch import nn from torch import nn, einsum
import torch.nn.functional as F import torch.nn.functional as F
import numpy as np import numpy as np
from functools import partial from functools import partial