Skip to content

[Bug] LSTM duplicates parameters for each step in the timeseries #12446

@SebastianBoblest

Description

@SebastianBoblest

It seems that TVM duplicates parameters in LSTMs and GRUs for each step in the input series.
This behavior has occurred before in the past but was fixed.
Maybe this issue is related to the refactoring done in #12213 but I could not really locate the source of the error.
I use TVM on commit ecbe4ca

I attached a zip file containing

  1. a toy lstm model that takes an input with two steps.
  2. a sample script to generate C code for it with TVM
  3. the generated code.

If you look into module_0.txt you can clearly see that identical parameters are in there twice.

@zhang-yi-chi @mbrookhart @AndrewZhaoLuo @vdkhoi @UlrikHjort @MichaelJKlaiber
lstm_bug.zip
r

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions