search

Tsinghua-MARS-Lab / StateTransformer

209 stars 18 forks source link

Integrate the diffusion decoder and some code frame adjustment #113

Closed Shiduo-zh closed 1 year ago

Shiduo-zh commented 1 year ago
  1. move mtr source code from libs/model/mtr to libs/mtr
  2. improve decoders for both forward and generation.
  3. support em algorithm for diffusion decoder to generate k>1 keypoints.
larksq commented 1 year ago

All comments were resolved