Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Future prediction length #170

Open
MartinPerry opened this issue Oct 27, 2024 · 0 comments
Open

Future prediction length #170

MartinPerry opened this issue Oct 27, 2024 · 0 comments

Comments

@MartinPerry
Copy link

If the output prediction length is longer than input frames, it is solved via this code (in SimVP class and same logic is applied to WaST and TAU as well):

pred_y = []
d = aft_seq_length // pre_seq_length
m = aft_seq_length % pre_seq_length
            
cur_seq = batch_x.clone()
for _ in range(d):
	cur_seq = self.model(cur_seq)
    pred_y.append(cur_seq)

if m != 0:
	cur_seq = self.model(cur_seq)
    pred_y.append(cur_seq[:, :m])
            
pred_y = torch.cat(pred_y, dim=1) 

Doesn't this degrade quality, when input to the model are basically the predicted frames? So the model sees correct past data and later some predicted data affected by error?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant