Transformer encoding


I just simply tried to run the example given in the following link on how to use the model as a standalone acoustic model:

when it runs the line:

# retrieve logits
 logits = model(input_values).logits

it gets stuck there. According to the code there is only one sample in the batch, so I assumed that this line should run rather quickly but it just simply gets stuck.

Any suggestion?