Skip to content

About An implement of AutoEncoder, Transformer and DDPM with cross-attention for Meteorlogical Graph seq-seq prediction

License

Notifications You must be signed in to change notification settings

LindleyLin/Meteor-Pred

Repository files navigation

Meteor-Pred

About An implement of AutoEncoder, Transformer and DDPM with cross-attention for Meteorlogical Graph seq-seq prediction

Technique

  • Feature extracting using basic Autoencoder with ConvNeXtV2 backbone or BYOL (not experimented)
  • Implicit vector predicting using Transformer with both encoder and decoder
  • Image generating using DDPM with cross-attention (better quality) or Autoencoder
  • Warmup and cosine lr-scheduler
  • Preprocess scripts without data augmentation

Autoencoder Sample

A basic sample using Autoencoder

DDPM Sample

A basic sample using DDPM

About

About An implement of AutoEncoder, Transformer and DDPM with cross-attention for Meteorlogical Graph seq-seq prediction

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages