Compatible input data spatial extent and patch size

[SarahAlidoost]

Right now in SpatioTemporalModel, there is a check as:

        assert H % self.patch_size[1] == 0 and W % self.patch_size[2] == 0, (
            "H and W must be divisible by patch size"
        )
        assert T % self.patch_size[0] == 0, "T must be divisible by patch size"

Can we have alternative like padding mechanism to avoid assertion?