About training

[lgs00]

Thanks for sharing the code.
When I changed the resolution of the training video to [832,608],
h = torch.cat([h, hs.pop()], dim=1)
There is an error: RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 20 but got size 19 for tensor number 1 in the list. Whether the resolution of the training is fixed?

[Doubiiu]

Maybe, you can try to change the resolutions to the multiples of 64.

[lgs00]

I want to know how much memory is needed to train the model，80GB is still not enough.