get_flops函数

[zswzswzsw233]

在get_flops脚本的construct_input函数中，构建输入数据时，对于时序输入的模型，是不是修改一下N=6？例如改成12
def construct_input(input_shape):
rot = torch.eye(4).float().cuda().view(1, 1, 4, 4).expand(1,6,4,4)

intrins = torch.eye(3).float().cuda().view(1,1, 3, 3).expand(1,6,3,3)
input = dict(img_inputs=[
    torch.ones(()).new_empty((1, 6, *input_shape)).cuda(), rot,
    rot, intrins, intrins,
    torch.ones((1, 6, 3)).cuda(),
    torch.eye(3).float().cuda().view(1, 3, 3)
])
return input

[Rayn-Wu]

Hi, 同样的问题。请问您解决了吗？