PosEmbed bug when the T_out > T_in

[zuliangfang]

Here we have to ensure the T <= self.maxT.

But it is not satisfied when the T_out > T_in. so we will get error on PosEmbded of decoder.

Here the self.mem_shape[i][0] = T_in = self.maxT, but the actual T in PosEmbed is T_out. When T_out > T_in, we will get error.

[gaozhihan]

Thank you very much for pointing it out! it's fixed in #51.

earth-forecasting-transformer/src/earthformer/cuboid_transformer/cuboid_transformer.py

Lines 2365 to 2369 in 0303b37

	if self.hierarchical_pos_embed:
	self.hierarchical_pos_embed_l = nn.ModuleList([
	PosEmbed(embed_dim=self.mem_shapes[i][-1], typ=pos_embed_type,
	maxT=target_temporal_length, maxH=self.mem_shapes[i][1], maxW=self.mem_shapes[i][2])
	for i in range(self.num_blocks - 1)])