111412:

TransformerEncoderLayer


1.8.1’s version does not take any batch_first argument (ref TransformerEncoderLayer — PyTorch 1.8.1 documentation 115 ), if you want that you need to upgrate to 1.9.0