Tan_pytorch_segmentation/pytorch_segmentation/PV_MLP/SMLP.py

10 lines
250 B
Python
Raw Normal View History

2025-05-19 20:48:24 +08:00
from model.mlp.sMLP_block import sMLPBlock
import torch
from torch import nn
from torch.nn import functional as F
if __name__ == '__main__':
input=torch.randn(50,3,224,224)
smlp=sMLPBlock(h=224,w=224)
out=smlp(input)
print(out.shape)