search

amazon-science / earth-forecasting-transformer

Official implementation of Earthformer
Apache License 2.0
349 stars 58 forks source link

misc fix #39

Closed gaozhihan closed 1 year ago

gaozhihan commented 1 year ago
  1. layout manipulation in previous vis_earthnet_seq adapted from the official impl is awkward and fails when the seq_len is smaller than channels, e.g. seq_len = 1.
  2. Improve the gallery() for visualization EarthNet2021 sequences. Enable seq_len indivisible by ncols by padding the seq_len.
  3. Make out_seq adjacent to in_seq in layout_to_in_out_slice().