13 lines
315 B
Python
13 lines
315 B
Python
|
|
||
|
|
from model.backbone.ConTNet import ConTNet
|
||
|
|
import torch
|
||
|
|
from torch import nn
|
||
|
|
from torch.nn import functional as F
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
model = build_model(use_avgdown=True, relative=True, qkv_bias=True, pre_norm=True)
|
||
|
|
input = torch.randn(1, 3, 224, 224)
|
||
|
|
out = model(input)
|
||
|
|
print(out.shape)
|
||
|
|
|