jhhuang96 / ConvLSTM-PyTorch

ConvLSTM/ConvGRU (Encoder-Decoder) with PyTorch on Moving-MNIST
MIT License
383 stars 85 forks source link

fixbug: Fixing incorrect if-elif-else code #12

Open kuhsinyv opened 2 weeks ago

kuhsinyv commented 2 weeks ago

This error causes the model to be initialized using the parameters of ConvGRU regardless of whether the user passes in the "--convlstm" comand-line argument.