I inference the model in cpu mode. I changed to use the graph mode to quantify the model, but the erros generated:

Screenshot from 2021-12-13 15-56-252051×274 40.6 KB

isn’t the if branch supported？
def forward(self, x):
if x.ndim == 5:
return self.forward_time_series(x)
else:
return self.forward_single_frame(x)