I tried to reproduce the error and couldn’t using some random values as model arguments.
Also I had to remove the chomp layers, since the implementation is missing. Could you post the code for these layers and the arguments you are using?