from thop import profile
device = 'cuda:1'
model = Transformer().to(device)
input = torch.randn(1, 201, 192).to(device)
macs, params = profile(model, inputs=(input,))
print("macsparams:",macs,params)
from thop import profile
device = 'cuda:1'
model = Transformer().to(device)
input = torch.randn(1, 201, 192).to(device)
macs, params = profile(model, inputs=(input,))
print("macsparams:",macs,params)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
