A Bug in PrithviViT
#1
by
EnoMomo
- opened
Hi,
In PrithviViT forward_features method in line 421, there is a bug.
Line 421: num_tokens_per_frame = x.shape[1] // self.patch_embed.num_frames
But self.patch_embed doesn't have num_frames parameter.
This is also implemented differently in terratorch package. The same line is implemented as following:
num_tokens_per_frame = x.shape[1] // self.num_frames
Thanks @EnoMomo for pointing this out, I updated the code. We had a lot of modifications in TerraTorch that we did not move to this repo.
blumenstiel
changed discussion status to
closed